install common dependencies and restructure

Dockerfile → JupyterLab/Dockerfile

@@ -18,3 +18,13 @@ ENV PATH "${CONDA_DIR}/envs/${conda_env}/bin:${PATH}"
 # set to default
 ENV CONDA_DEFAULT_ENV ${conda_env}
 
+# install pytorch 1.9.1
+RUN conda install numpy=1.21.2 && \
+    conda install pytorch=1.9.1 torchvision  cudatoolkit=10.2 -c pytorch && \
+    conda clean --all -f -y
+
+# install pip requirements
+COPY requirements.txt /tmp/
+RUN "${CONDA_DIR}/envs/${conda_env}/bin/pip" install --quiet --no-cache-dir -r /tmp/requirements.txt
+
+

JupyterLab/requirements.txt

+pytorch-lightning
+matplotlib
+requests
+pyyaml
+pandas
+scipy
+scikit-learn
+seaborn
+  
\ No newline at end of file

requirements.txt

-torch==1.9.1
-torchvision
\ No newline at end of file