AssertionError：Torch not compiled with CUDA enabled

最新推荐文章于 2024-04-11 10:20:13 发布

从此烟雨落金城~

最新推荐文章于 2024-04-11 10:20:13 发布

阅读量4.4k

点赞数 2

文章标签： python pytorch pycharm

本文链接：https://blog.csdn.net/weixin_44340927/article/details/128297930

版权

安装完torch、torchvision后，报错：AssertionError：Torch not compiled with CUDA enabled

一、打开 anaconda powershell prompt，输入如下命令：

>python
>>>import torch
>>>a = torch.ones((3,1))   //创建一个长度为 3 值为 1 的矩阵
>>>a = a.cuda(0)              //将该矩阵挪到 cuda 0 也即时本机的 GPU 上

若如上命令行执行过程中均无报错则表明本机上成功安装
了 torch 与对应的 torchvision。

二、安装torch和torchvision后，还是报错：AssertionError：Torch not compiled with CUDA enabled

在执行解决方法一最后的命令行时，发生如下报错：
在这里插入图片描述
这种情况很大可能是因为 torch 与 CUDA 的版本不兼容导致的，为了验证这一假设，可以在终端输入如下命令：

>>>print(torch.cuda.is_available())

若输出结果为False的话，就代表引起报错的原因是torch与CUDA的版本不兼容。
在这里插入图片描述

先输入：

exit()

退出python环境
在这里插入图片描述

接下来查看CUDA的版本，在终端输入：
nvcc -V

输出结果如下：
在这里插入图片描述
可以看到我本机的 CUDA 版本是 10.0

去 Pytorch 的官网，找到与你本机 CUDA 对应版本的 Pytorch，并进行安装。
在这里插入图片描述

在pyCharm的terminal中输入：

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 -c pytorch

在这里插入图片描述

关注