如果本博客对您有用。感谢您的点赞,收藏,关注。
使用paddle的device的get和set来获取和设置:
import paddle
print(paddle.device.get_device())
如果有GPU正在运行的话,运行结果如下:(注意,它只能显示正在运行的GPU)
如果你想查看主机有几个GPU,可以终端输入nvidia-smi查看
上图显示,0号GPU在运行。
之后在代码开始设置 指定 选择GPU的设备。通过set_device()方法设置设备即可:
paddle.device.set_device('gpu:0')
之后就可以用其他内容了
例如:model = MNIST()
train(model)
.
如果你有多个GPU:0,1,2,3,4,5,6......
指定某个GPU 可以通过上面的set_device()方法指定。
如果本博客对您有用。感谢您的点赞,收藏,关注。
如果以上不能实现GPU的指定,选择训练。可以参考我亲测的方法
这种方法。针对windows下的GPU设置非常使用。本人亲测可用。