AssertionError: Torch not compiled with CUDA enabled

Posted 2023-03-02 followtheheart

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了AssertionError: Torch not compiled with CUDA enabled相关的知识，希望对你有一定的参考价值。

在用pycharm跑深度学习的一段代码时出现了这样的错误。
原因是cpu代码不能跑gpu的程序，解决办法如下：
在代码前加入

`device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

`再把代码中的

.cuda()

替换成

 .to(device)

以上是关于AssertionError: Torch not compiled with CUDA enabled的主要内容，如果未能解决你的问题，请参考以下文章