docker update

This commit is contained in:
2023-06-06 22:17:49 +02:00
parent 05341fd48f
commit cac382ef19
2 changed files with 3 additions and 3 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
# Use an existing image as the base
# FROM nvcr.io/nvidia/pytorch:23.04-py3
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
# FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
# Add the current user to the image