torch GPU 在cmd 中可以运行成功。 但是在pycharm中不能运行。

以下代码
print(torch.version)
print(torch.cuda.is_available())
print(torch.zeros(1).cuda())
cmd 中可以运行成功。
但是在pycharm中不能运行。

在这里插入图片描述

在这里插入图片描述

吗的我也是,cmd conda一个环境下,用python命令 python一个文件,cuda.count是0,进入python 再看cuda.count就是1了

img

你的x,y后面有没有加cuda()呢?

pycharm选择的编译器版本不对

确认一下你的torch框架是否装在了你的虚拟环境:python39中。