Commit 6d4c72b6 authored by Thomas Lips's avatar Thomas Lips
Browse files

torch 1.10

parent 1dfe1581
Pipeline #52515 passed with stages
in 38 minutes and 8 seconds
...@@ -25,7 +25,7 @@ ENV CONDA_DEFAULT_ENV ${conda_env} ...@@ -25,7 +25,7 @@ ENV CONDA_DEFAULT_ENV ${conda_env}
# install pytorch 1.9.1 # install pytorch 1.9.1
RUN conda install numpy=1.21.2 && \ RUN conda install numpy=1.21.2 && \
conda install pytorch=1.9.1 torchvision cudatoolkit=10.2 -c pytorch && \ conda install pytorch=1.10 torchvision cudatoolkit=10.2 -c pytorch && \
conda clean --all -f -y conda clean --all -f -y
# install pip requirements # install pip requirements
......
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