38,757
edits
No edit summary |
(Updating to match new version of source page) |
||
Line 487: | Line 487: | ||
<div class="mw-translate-fuzzy"> | |||
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 | ||
</div> | |||
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. |