在GPU上创建Tensor张量时出现错误：Torch not compiled with CUDA enabled 解决。

当我在pycharm中在GPU上创建Tensor张量时出现了错误：

import torch
dev = torch.device('cuda:0')
a = torch.tensor([2,2],dtype = torch.float32 , device = dev)
print(a)

问题如下：

该问题出现的原因：您安装的pytorch版本与你的cuda对应不上，导致不能成功创建张量。

检验方式如下：

python  
import torch  
torch.cuda.is_available()  

如果出现的为false则代表确实是您的版本对应不上。

请先把你安装好的pytorch先卸载干净在重新安装。如果你是在conda下安装的则卸载指令如下：

conda remove pytorch
conda remove pytorchvision

之后就能进行安装了。

安装网址及方式可以参照这篇文章：Ubuntu20.04版本安装pytorch（宝宝级攻略）