0
点赞
收藏
分享

微信扫一扫

多块GPU的使用

拾杨梅记 2022-03-11 阅读 47

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

相关推荐

0 条评论