(windows/ubuntu系统)paddlepaddle 查看 指定 选择GPU进行训练

如果本博客对您有用。感谢您的点赞，收藏，关注。

使用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的指定，选择训练。可以参考我亲测的方法

https://blog.csdn.net/Vertira/article/details/122081143https://blog.csdn.net/Vertira/article/details/122081143

这种方法。针对windows下的GPU设置非常使用。本人亲测可用。