多块GPU的使用

多块GPU的使用

python -m torch.distributed.launch --nproc_per_node=3 --use_env train_multi_GPU.py --resume '/home/.../model4.pth'

单块GPU的使用

python train_ssd300.py --device cuda:'7'

指定特定的 GPU进行训练

在程序运行的命令前加上  CUDA_VISIBLE_DEVICE=0,2,3  命令

还没有太理解

查看cuda版本

nvidia-smi

conda 环境的创建

conda create -n jinssd python=3.7

你可能感兴趣的:(pytorch,深度学习,python)