Skip to content

Commit 48aa33a

Browse files
committed
Upgrade PyTorch
1 parent fe3a914 commit 48aa33a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.devcontainer/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ RUN apt-get update \
4040
&& pip --disable-pip-version-check --no-cache-dir install scikit-learn \
4141
&& pip --disable-pip-version-check --no-cache-dir install scikit-image \
4242
&& pip --disable-pip-version-check --no-cache-dir install scipy \
43-
&& pip --disable-pip-version-check --no-cache-dir install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html \
43+
&& pip --disable-pip-version-check --no-cache-dir install torch==1.6.0+cpu torchvision==0.7.0+cpu -f https://download.pytorch.org/whl/torch_stable.html \
4444
&& pip --disable-pip-version-check --no-cache-dir install tensorflow \
4545
#
4646
# Create a non-root user to use if preferred - see https://aka.ms/vscode-remote/containers/non-root-user.

0 commit comments

Comments
 (0)