luni/reconstruction
1
0
Fork 0
Code Issues 6 Pull Requests Packages Projects 1 Releases Wiki Activity

update docker

Browse Source
This commit is contained in:
luni
2023-05-09 23:04:01 +00:00
parent 07facfe064
commit 3f60403167
3 changed files with 25 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docker/Dockerfile
+3 -2
View File
@@ -1,6 +1,7 @@
# Use an existing image as the base
#FROM nvcr.io/nvidia/pytorch:22.06-py3
FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
# FROM nvcr.io/nvidia/pytorch:23.04-py3
FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
# FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
# Add the current user to the image
ARG USER_NAME=lhunor