AssertionError: CUDA unavailable, invalid device cuda 0 requested

谁践踏了优雅 2022-11-17 05:25 663阅读 0赞

先单独创建一个py文件用于测试cuda是否配置正确

  1. print(torch.cuda.is_available())

运行后输出的是True,这代表你的cuda已经安装好了。
打开报错的程序,在主程序里面加入上面那一句:

  1. print(torch.cuda.is_available())

然后把device改成cuda,默认是cuda0
然后运行问题就解决了。

发表评论

表情:
评论列表 (有 0 条评论,663人围观)

还没有评论,来说两句吧...

相关阅读