lingyuzeng/nvidia_docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Your Name
2024-08-02 16:39:13 +08:00
parent 7b48198f1d
commit 5fca5e7564
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
docker-compose_colossalai.yml
View File

@@ -6,10 +6,10 @@ services:
context: . context: .
dockerfile: Dockerfile.colossalai dockerfile: Dockerfile.colossalai
args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822 args: # PyTorch版本、Python版本与pytorch_lightning版本的对应关系表 https://blog.csdn.net/qq_41813454/article/details/137421822
REGISTRY: "nvcr.io" # REGISTRY: "nvcr.io"
OWNER: "nvidia" # nvcr.io/nvidia/pytorch:24.07-py3 # OWNER: "nvidia" # nvcr.io/nvidia/pytorch:24.07-py3
LABEL: "pytorch" # LABEL: "pytorch"
VERSION: "24.07-py3" # VERSION: "24.07-py3"
# HTTP_PROXY: "http://127.0.0.1:15777" # HTTP_PROXY: "http://127.0.0.1:15777"
# HTTPS_PROXY: "http://127.0.0.1:15777" # HTTPS_PROXY: "http://127.0.0.1:15777"
CACHEBUST: 1 CACHEBUST: 1