Pytorch错误:AssertionError: Torch not compiled with CUDA enabled

在这里插入图片描述
windows上已经安装过显卡驱动,可在终端通过nvcc -V命令查看
Pytorch错误:AssertionError: Torch not compiled with CUDA enabled_第1张图片
在代码中检查cuda是否启用:

print(torch.cuda.is_available())

如果输出的结果是False,那么说明当前的Pytorch版本无法使用显卡
需要重新安装GPU版的Pytorch
可参考以下博客的安装过程
windows10 Anaconda安装pytorch(GPU)踩坑!

你可能感兴趣的:(Pytorch错误:AssertionError: Torch not compiled with CUDA enabled)