This commit is contained in:
Your Name
2024-07-17 06:14:32 +00:00
parent e7e28fd76d
commit 7f7d490de6
2 changed files with 4 additions and 3 deletions

View File

@@ -27,6 +27,7 @@ services:
OWNER: "nvidia" # nvcr.io/nvidia/pytorch:24.06-py3
LABEL: "pytorch"
VERSION: "24.06-py3"
NV_PEER_MEM_VERSION: "1.2"
DS_BUILD_OPS: 1
DEEPSPEED_VERSION: "master"
DEEPSPEED_INSTALL_FLAGS: "--allow_sudo"