选择第三张卡运行，如果不指定，默认选择第一张卡进行运算，如果第一张卡在运行其他程序则不能运行即出现内存溢出情况。
CUDA_VISIBLE_DEVICES=2 python train.py
远程或者本地服务器指定某张卡运行
猜你喜欢
转载自blog.csdn.net/DENGSHUCHAO152/article/details/123325322
今日推荐
周排行