2.0.1-cuda11.7-cudnn8-devel

Signed-off-by: siaimes <[email protected]>
siaimes · Jun 23, 2023 · fca23d7 · fca23d7
1 parent cdf9f6f
commit fca23d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 
 RUN apt-get update && apt-get install -y libgl1-mesa-glx libpci-dev curl nano psmisc zip git && apt-get --fix-broken install -y