(1)先查看当前服务器下的显卡情况:nvidia-smi
(2)用如下命令指定某张卡,来运行程序: CUDA_VISIBLE_DEVICES=4 python main.py --cuda
本文标题:cuda指定某块gpu运行程序
本文链接:https://www.haomeiwen.com/subject/twfivctx.html
网友评论