1.8.1-cuda11.1-cudnn8-devel

siaimes · siaimes · commit c1c86e1d6921 · 2023-06-23T23:58:58.000+08:00
Signed-off-by: siaimes &lt;34199488+siaimes@users.noreply.github.com&gt;
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel
+FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
 
 RUN apt-key del 7fa2af80 && \
     apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM pytorch/pytorch:1.7.1-cuda11.0-cudnn8-devel`
	`1`	`+FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel`
`2`	`2`
`3`	`3`	`RUN apt-key del 7fa2af80 && \`
`4`	`4`	`apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub && \`