AssertionErrorTorch not compiled with CUDA enabled

安装完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

这篇好文章是转载于：学新通技术网