在用pycharm跑深度学习的一段代码时出现了这样的错误。
原因是cpu代码不能跑gpu的程序,解决办法如下:
在代码前加入
`device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
`再把代码中的
.cuda()
替换成
.to(device)
在用pycharm跑深度学习的一段代码时出现了这样的错误。
原因是cpu代码不能跑gpu的程序,解决办法如下:
在代码前加入
`device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
`再把代码中的
.cuda()
替换成
.to(device)