1.6.0-cuda10.1-cudnn7-devel

Signed-off-by: siaimes <34199488+siaimes@users.noreply.github.com>
siaimes · Jun 23, 2023 · 396ea56 · 396ea56
1 parent d247abc
commit 396ea56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.5.1-cuda10.1-cudnn7-devel
+FROM pytorch/pytorch:1.6.0-cuda10.1-cudnn7-devel
 
 RUN apt-key del 7fa2af80 && \
     apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \