python报错系列(5)--AssertionError: Torch not compiled with CUDA enabled
失败是迷雾,穿过它,我们就可以瞥见光明
·
前言
1、AssertionError: Torch not compiled with CUDA enabled
此错误是由于下载的torch没有cuda,在运行时就会出错;
2、解决方式
在程序最开始的地方加上
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
举例如下:
代码其余地方出现.cuda()的地方改成.to(device)就可以在无gpu的环境中运行了
总结
分享:
Don"t makepromisesyou can"t keep.But those are the bestkind.不轻许做不到的承诺,但做不到的承诺往往最美好。
更多推荐
已为社区贡献4条内容
所有评论(0)