Updated version of the base image

9 months ago · 1019666122
parent 21caf98d27
commit 1019666122
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
 ENV TZ=Europe/Samara
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone