This commit is contained in:
2024-06-21 13:46:40 +08:00
parent 5aeb76da25
commit 6bd77c71ca
4 changed files with 35 additions and 33 deletions

View File

@@ -6,6 +6,7 @@ services:
context: .
dockerfile: Dockerfile
args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822
PYTHON_VERSION: 3.9
CUDA_VERSION: 11.7.1
PYTORCH_VERSION: 1.13.1
TORCHVISION_VERSION: 0.14.1