raise AssertionError(“Torch not compiled with CUDA enabled“)

raise AssertionError(“Torch not compiled with CUDA enabled”)

遇到的问题,就是 GPU driver 以及cuda都装完了,pytorch装完之后想试试代码。
结果报错,raise AssertionError(“Torch not compiled with CUDA enabled”)。

解决方案

就是打开pytorch 官网,对照一下 cuda 版本和 pytorch版本是不是对的上。
比如我的 cuda 版本是 11.1, 那么 装 1.8.0 是可行的

另外,我再 cuda 11.0 的情况下,也试着装了一下 pytorch 1.8.1,
结果也是能够调用 GPU的,就是不知道会不会有隐患。
总之遇到这个问题,可以考虑一下 cuda 版本和 pytorch版本 能不能在下载官网上找到对应的,然后修改一下。

例子:https://github.com/pytorch/pytorch/issues/30664

pytorch 官网: https://pytorch.org/get-started/previous-versions/

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年5月9日
下一篇 2023年5月9日

相关推荐