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。
下载对应的CUDA
输入网址https://developer.nvidia.com/cuda-toolkit-archive,这里我下载11.6.0版本的CUDA。
接着进入如下界面,进行如下选择。
然后开始安装。
然后直接全部下一步。查看是否安装成功,命令行 输入nvcc -V,成功显示如下,则安装成功
下载对应的cuDNN
可以参考博客:https://blog.csdn.net/jhsignal/article/details/111401628
二:查看与自己的cuda对应的版本
进入官网https://pytorch.org/get-started/previous-versions/
复制该命令到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