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