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/