python代码放gpu跑,python指定gpu

GPU运行python代码 步骤 1.查看电脑的显卡有几块(在控制台输入) nvidia-smi

出现以下内容

这里只有0号显卡

2.在代码中指定GPU import osos.environ["CUDA_VISIBLE_DEVICES"] = "0"

这样就可以啦！！！！

历史中提交的图片或压缩文件