耗时数天,今天终于解决了
狂喜!!!
如果你目标是在虚拟环境中安装Pytorch,
如果你确定你的CUDA版本和CUDNN版本、torch版本、torchaudio版本、torchvision版本对应,
但当你在虚拟环境中ipython然后import torch然后torch.cuda.is_available()后仍然返回false,
那么请检查一下在虚拟环境外你是否有安装torch,检查下那个torch是否是cpu版本,因为我发现,我在虚拟环境中运行ipython然后import torch后,它好像import的是虚拟环境外的torch,而我虚拟环境外的刚好是cpu版本,所以无论怎么变虚拟环境内的pytorch它都返回false
因此,我先删除虚拟环境外的Pytorch的包,然后参考了以下链接的方法:torch.cuda.is_available()返回false——解决办法_cuda available false_Nefu_lyh的博客-CSDN博客jhttps://blog.csdn.net/qq_46126258/article/details/112708781
把在虚拟环境里Pytorch的几个包复制到了虚拟环境外,结果再在虚拟环境中ipython然后import torch然后torch.cuda.is_available()后,显示True,结果如下图:
但是,我后来发现ipython后环境好像换到了base环境下,所以我在虚拟环境中输入python后再import torch然后torch.cuda.is_available(),最后返回依然为True
不过其间也遇到了一些小bug:
第一个是ModuleNotFoundError: No module named 'typing_extensions',这个直接conda install typing_extensions就好。
还有一个numpy方面的报错,我直接conda install numpy就好了。
另外,附上我安装的包的版本:
我的驱动版本:
安装CUDA、CUDNN、Pytorch的过程中也有很多其他可能踩坑的点,比如以下几点:
1.安装pytorch的命令应该pip而不是conda(这个图就浅浅借一下链接作者的吧~~)
2.pip时报错:首先在命令行操作时不能挂梯子,其次还报错的话应该先把三个whl包下载到本地再pip
pip install torch-2.0.1+cu118-cp311-cp311-win_amd64.whl
pip install torchaudio-2.0.2+cu118-cp311-cp311-win_amd64.whl
pip install torchvision-0.15.2+cu118-cp311-cp311-win_amd64.whl
以上一些点其他作者也写过很多,这里就不过多赘述了
最后祝大家配环境的过程中少踩坑,少遇bug!!!