多块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