fix: use ubuntu 18.04 base image in dlc gpu image

Dan Choi · ChoiByungWook · commit 66b6f166329e · 2020-12-10T19:52:29.000-08:00
diff --git a/test/container/1.6.0/Dockerfile.dlc.gpu b/test/container/1.6.0/Dockerfile.dlc.gpu
@@ -1,5 +1,5 @@
 ARG region
-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
 
 COPY dist/sagemaker_pytorch_training-*.tar.gz /sagemaker_pytorch_training.tar.gz
 RUN pip install --upgrade --no-cache-dir /sagemaker_pytorch_training.tar.gz && \
