This commit is contained in:
Jon Shelley 2023-06-23 14:23:37 -06:00 коммит произвёл GitHub
Родитель d05eb2e6cf
Коммит 96f266216a
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 3 добавлений и 2 удалений

Просмотреть файл

@ -1,7 +1,8 @@
#!/bin/bash
HPCX_DIR=`ls /opt | grep -i hpcx`
CONT="nvcr.io#nvidia/pytorch:21.09-py3"
MOUNT="/shared/data/azure/benchmarking/NDv4/cc-slurm-ngc/nccl:/nccl,/opt/hpcx-v2.9.0-gcc-MLNX_OFED_LINUX-5.4-1.0.3.0-ubuntu18.04-x86_64:/opt/hpcx"
MOUNT="/shared/data/azure/benchmarking/NDv4/cc-slurm-ngc/nccl:/nccl,/opt/${HPCX_DIR}:/opt/hpcx"
export OMPI_MCA_pml=ucx
export OMPI_MCA_btl=^openib