Commit 313b07bf authored by Thomas Lips's avatar Thomas Lips
Browse files

install common dependencies and restructure

parent 2321fc35
......@@ -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
pytorch-lightning
matplotlib
requests
pyyaml
pandas
scipy
scikit-learn
seaborn
\ No newline at end of file
torch==1.9.1
torchvision
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment