win11:Torch not compiled with CUDA enabled

时间:2021-10-20 01:18:12

1、我的环境

系统为win11,使用anaconda环境,python版本3.8,已经通过anaconda安装了pytorch 2.0,但是通过命令行nvcc -V以及文件查看发现未安装CUDA

整体思路:先安装的pytorch2.0,然后发现未安装CUDA,安装自己电脑对应的CUDA即11.6版本后,再次运行python程序,发现仍然报错Torch not compiled with CUDA enabled,就去官网https://pytorch.org/get-started/previous-versions/ 查看与自己的cuda对应的版本,卸载当前安装的pytorch,使用官网给出的命令下载pytorch。

一:安装自己电脑对应的CUDA版本

查看CUDA版本

首先桌面上右键->点击选择更多选项->点击NAVID控制面板
打开后:点击帮助,系统信息,组件。这里我的CUDA版本最大支持到11.6.134,我可以安装的CUDA版本不能大于11.6.134,这时我就需要去官网下载11.6.134版本以下的CUDA。
win11:Torch not compiled with CUDA enabled

下载对应的CUDA

输入网址https://developer.nvidia.com/cuda-toolkit-archive,这里我下载11.6.0版本的CUDA。
win11:Torch not compiled with CUDA enabled
接着进入如下界面,进行如下选择。
win11:Torch not compiled with CUDA enabled
然后开始安装。
win11:Torch not compiled with CUDA enabled
然后直接全部下一步。查看是否安装成功,命令行 输入nvcc -V,成功显示如下,则安装成功
win11:Torch not compiled with CUDA enabled

下载对应的cuDNN

可以参考博客:https://blog.csdn.net/jhsignal/article/details/111401628

二:查看与自己的cuda对应的版本

进入官网https://pytorch.org/get-started/previous-versions/
win11:Torch not compiled with CUDA enabled
复制该命令到anaconda的命令里,安装完毕后在anaconda的python环境输入

import torch
print(torch.cuda.isavailable())

如果成功则显示True,问题就解决了。

三:其他可能的问题

当然我查找问题的时候发现可能存在其他的问题,如:
1,你的pytorch是否安装错了,没有安装GPU版本的
2,anaconda的pytorch和本地的pytorch冲突,参考 https://blog.csdn.net/qq_46941656/article/details/119701547