$ nvidia-smi
查看当前节点下有几块GPU，以及每一块的使用情况
在运行程序时指定使用哪块GPU
$ CUDA_VISIBLE_DEVICES=5 python main.py --cuda
cuda指定使用哪块GPU
猜你喜欢
转载自blog.csdn.net/ccbrid/article/details/79761674
今日推荐
周排行