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
+2 -2
View File
@@ -40,8 +40,8 @@ services:
- capabilities: [gpu]
lhunor_torch_3:
image: lhunor_reconstruction:v2.0
container_name: lhunor_reconstruction_v2.0
image: lhunor_reconstruction:v2.0.1
container_name: lhunor_reconstruction_v2.0.1
ipc: host
tty: true
ulimits: