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

Question

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

Answer 1

你的x，y后面有没有加cuda()呢？