We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f5944f3 commit 66b6f16Copy full SHA for 66b6f16
test/container/1.6.0/Dockerfile.dlc.gpu
@@ -1,5 +1,5 @@
1
ARG region
2
-from 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu110-ubuntu16.04
+from 763104351884.dkr.ecr.$region.amazonaws.com/pytorch-training:1.6.0-gpu-py36-cu110-ubuntu18.04
3
4
COPY dist/sagemaker_pytorch_training-*.tar.gz /sagemaker_pytorch_training.tar.gz
5
RUN pip install --upgrade --no-cache-dir /sagemaker_pytorch_training.tar.gz && \
0 commit comments