rsnt_translations
56,420
edits
(Updating to match new version of source page) |
No edit summary |
||
Line 487: | Line 487: | ||
module load StdEnv/2020 | |||
module load python/3.8 | module load python/3.8 | ||
virtualenv --no-download $SLURM_TMPDIR/env | virtualenv --no-download $SLURM_TMPDIR/env | ||
source $SLURM_TMPDIR/env/bin/activate | source $SLURM_TMPDIR/env/bin/activate | ||
pip install --no-index tensorflow==2.5.0 horovod | pip install --no-index tensorflow==2.5.0 horovod | ||
export NCCL_BLOCKING_WAIT=1 #Set this environment variable if you wish to use the NCCL backend for inter-GPU communication. | export NCCL_BLOCKING_WAIT=1 #Set this environment variable if you wish to use the NCCL backend for inter-GPU communication. |