AssertionError: Torch not compiled with CUDA enabled
解决问题:AssertionError: Torch not compiled with CUDA enabled
·

简单解决方法:
在代码前添加如下代码:
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

再将代码其余地方出现.cuda()的地方改成.to(device),就可以在无gpu的环境中运行了。
例如:


不简单的:
代码本身可以确定是没有问题的,报错含义是在编译torch的时候,CUDA并没有起作用。
检查Anaconda prompt 的基本环境中,是否装有Pytorch(GPU版)。

经检查,在Anaconda的基本环境中没有安装Pytorch(GPU版),所以Pycharm即使引入了conda environment,也没有办法使用Pytorch(GPU版)。
解决:安装Pytorch(GPU版)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

还是失败了 T-T。
更多推荐
https://blog.csdn.net/dujuancao11/article/details/114006234


所有评论(0)