如何再GPU上运行py代码？

Question

本人用spyder编写了一个py代码，在cpu上能够正常运行。为了实现在GPU上运行，在代码头部直接插入下面两行代码：
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
然后运行程序发现还是在CPU上运行，网上好多说是GPU编号的问题，本人的电脑是集显加独显，
上面只有编号为0的GPU，但是不管后面改成0还是1都没有效果。
然后尝试在终端中直接运行这个py程序，网上说的代码是CUDA_VISIBLE_DEVICES=1，python +py代码，我试了一下，发现会报错，

随后又搜了一番解决方法，有人说是环境变量的问题，我的环境变量应该是对的，

在deviceQuery.exe，bandwidthTest.exe测试都能通过，但为什么就是不行呢，求大佬解答！

Answer 1

你有把cudnn放到cuda目录下吗？