在运行项目时,突然报错AssertionError: Torch not compiled with CUDA enabled
可以先通过以下代码查看是否安装了GPU版本的torch
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果输出是False则大概率表明未安装GPU版本的torch,这是只要去官网下载一个GPU版本的torch即可
在运行项目时,突然报错AssertionError: Torch not compiled with CUDA enabled
可以先通过以下代码查看是否安装了GPU版本的torch
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果输出是False则大概率表明未安装GPU版本的torch,这是只要去官网下载一个GPU版本的torch即可