update docker

2023-04-27 19:01:21 +00:00
parent 77485b5a31
commit 43752d085a
5 changed files with 27 additions and 24 deletions
@@ -1,5 +1,6 @@
 # Use an existing image as the base
-FROM nvcr.io/nvidia/pytorch:22.06-py3
+#FROM nvcr.io/nvidia/pytorch:22.06-py3
+FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel

 # Add the current user to the image
 ARG USER_NAME=lhunor