docker update
This commit is contained in:
+1
-1
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user