docker update
This commit is contained in:
+1
-1
@@ -1,6 +1,6 @@
|
|||||||
# Use an existing image as the base
|
# Use an existing image as the base
|
||||||
# FROM nvcr.io/nvidia/pytorch:23.04-py3
|
# 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
|
# FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
|
||||||
|
|
||||||
# Add the current user to the image
|
# Add the current user to the image
|
||||||
|
|||||||
@@ -40,8 +40,8 @@ services:
|
|||||||
- capabilities: [gpu]
|
- capabilities: [gpu]
|
||||||
|
|
||||||
lhunor_torch_3:
|
lhunor_torch_3:
|
||||||
image: lhunor_reconstruction:v2.0
|
image: lhunor_reconstruction:v2.0.1
|
||||||
container_name: lhunor_reconstruction_v2.0
|
container_name: lhunor_reconstruction_v2.0.1
|
||||||
ipc: host
|
ipc: host
|
||||||
tty: true
|
tty: true
|
||||||
ulimits:
|
ulimits:
|
||||||
|
|||||||
Reference in New Issue
Block a user