Upgrade base docker image

takuseno · Oct 21, 2023 · c3d834e · c3d834e
1 parent c5d50aa
commit c3d834e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,11 +1,10 @@
-FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
 
 # this needs to avoid time zone question
 ENV DEBIAN_FRONTEND=noninteractive
 
 # install dependencies
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \
-    apt-get update && \
+RUN apt-get update && \
     apt-get install -y --no-install-recommends \
       build-essential \
       software-properties-common \