无法在GPU上运行/Cannot run on GPU
FrancisGrace opened this issue · comments
❓ Questions & Help
我的电脑有一个GPU,我在experiments()
函数中指定了参数devices=[0]
,但运行后CPU利用率接近100%,GPU利用率几乎为0,显然程序并未在GPU上运行。
My computer has a GPU, and I declared the argument devices=[0]
in the function experiments()
. However, when I ran it, the utilization of CPU reached almost 100% while that of GPU was still almost 0%. Obviously the program didn't run on the GPU.
Hi @FrancisGrace,
请问你是否成功安装了gpu版本的pytorch?可以通过输出torch.cuda.is_available()检查一下。
Hi @FrancisGrace,
请问你是否成功安装了gpu版本的pytorch?可以通过输出torch.cuda.is_available()检查一下。
@cenyk1230 是的,我的PyTorch版本为1.11,虚拟环境下这个函数返回值为True