PyTorch训练(十):进行多卡训练【CUDA_VISIBLE_DEVICES=‘0,2,3‘ python main.py】【前提是代码中已经设置好了DP或DDP多卡训练模式】

u013250861 2022-09-09 02:57:10 阅读数:653

Pytorch进行训练cuda_visible_devices

PyTorch训练(十):进行多卡训练【CUDA_VISIBLE_DEVICES=‘0,2,3’ python main.py】【前提是代码中已经设置好了DP或DDP多卡训练模式】

版权声明:本文为[u013250861]所创,转载请带上原文链接,感谢。 https://blog.csdn.net/u013250861/article/details/126770517