This commit is contained in:
Your Name
2024-07-17 07:09:16 +00:00
parent 7f7d490de6
commit edfdd28619
2 changed files with 31 additions and 17 deletions

View File

@@ -27,7 +27,6 @@ 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"