CPU显卡运行报错AssertionError Torch not compiled with CUDA enabled解决

本来代码里有一句device = ('cuda'if torch.cuda.is_available else 'cpu')

即用CUDA计算,否则用CPU。

但是一直报错AssertionError Torch not compiled with CUDA enabled

查了半天也查不到确切的问题,一怒之下把代码改成了device=('cpu')

好了。

贴上觉得有用的其他帖子。

https://blog.csdn.net/weixin_41243949/article/details/103363233

https://blog.csdn.net/qq_45504119/article/details/105405151

https://blog.csdn.net/m0_63483436/article/details/124122014

(为啥要说我文章质量低!?)

你可能感兴趣的:(深度学习,python,pytorch)