文章讨论了在使用PyTorch时遇到的CUDA相关问题，包括`torch.cuda.is_available`返回假，AssertionError关于未编译CUDA，以及RuntimeError由于找不到设备执行的内核。提供了针对这些问题的解决方案，如检查并安装匹配的PyTorch和CUDA版本，推荐使用conda进行安装。
问题1.torch.cuda.is_available始终返回false
原因：pytorch与CUDA版本不匹配
解决方案：参考https://download.pytorch.org/whl/torch_stable.html ，找到自己对应的版本。
例如，cu117代表CUDA11.7，cp39代表python3.9。
问题2.AssertionError: Torch not compiled with CUDA enabled
原因：同样是pytorch与CUDA版本不匹配
解决方案：参考上个问题，重新安装环境。建议使用conda， conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch.
其中11.7即为CUDA版本号，根据自己情况修改。
问题3.RuntimeError: CUDA error: no kernel image is available for execution on the device
参考http://t.csdnimg.cn/beFGM
扫一扫
467
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
