先
nvidia-smi
看看使用情况
如:
再
export CUDA_VISIBLE_DEVICES=0,1,2
指定使用的 GPU
Cold_Chair ©著作权
先
nvidia-smi
看看使用情况
如:
再
export CUDA_VISIBLE_DEVICES=0,1,2
指定使用的 GPU
下一篇:用 SSH 给服务器网络加速
1.强制只使用cpu: import os#os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" # see issue #152os.environ["CUDA_VISIBLE_DEVICES"] = "" 注意:os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"可能会改变没有次句时GPU的默认序号。 2.
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M