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/