原博:https://blog.csdn.net/lily1234567/article/details/81945892
我同样遇到这个问题,并且显卡显示
照博客的方法在程序中添加了代码:
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
问题解决。
后来又出现如下的问题,参考博客解决了 希望这次没问题了
https://blog.csdn.net/strive_for_future/article/details/81292865