Skip to content

Commit a4c13e0

Browse files
committed
fixed pytorch version
1 parent 1b10fe9 commit a4c13e0

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

vendor/ngc-pytorch/Dockerfile.21.12-py3

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -107,6 +107,7 @@ RUN git clone -q --branch=v0.3.18 git://github.com/xianyi/OpenBLAS.git && \
107107
cd OpenBLAS && \
108108
make DYNAMIC_ARCH=1 NO_AFFINITY=1 NUM_THREADS=48 FC=gfortran && \
109109
make install && \
110+
cd /tmp && \
110111
git clone --recursive https://github.com/bodono/scs-python.git && \
111112
cd /tmp/scs-python && \
112113
python setup.py install --scs --gpu
@@ -188,7 +189,7 @@ RUN apt autoclean && \
188189

189190
RUN /opt/conda/bin/python3 -m ipykernel install \
190191
--prefix=/opt/conda/ \
191-
--display-name "PyTorch 1.10 (NGC 21.12/Python 3.8 Conda) on Backend.AI" && \
192+
--display-name "PyTorch 1.11 (NGC 21.12/Python 3.8 Conda) on Backend.AI" && \
192193
cat /opt/conda/share/jupyter/kernels/python3/kernel.json
193194

194195
# Backend.AI specifics

0 commit comments

Comments
 (0)