Commit 3db9583c authored by Thomas Lips's avatar Thomas Lips
Browse files

update to torch 1.10

parent 6d4c72b6
Pipeline #52558 passed with stages
in 14 minutes and 12 seconds
......@@ -23,7 +23,7 @@ ENV CONDA_DEFAULT_ENV ${conda_env}
## install dependencies
# install pytorch 1.9.1
# install pytorch 1.10
RUN conda install numpy=1.21.2 && \
conda install pytorch=1.10 torchvision cudatoolkit=10.2 -c pytorch && \
conda clean --all -f -y
......
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