由于笔者自己在下载pytorch的时候,踩过诸多坑。一直想着出一套完整的教程,但是时间久远,很多之前保存的下载过程图片都没有了。
特写此篇,来进行一下小小的参考。(第一次写,请见谅~(抱拳!))
一、win10安装CUDA+cuDNN
首先要查看自己电脑上的NVIDIA、CUDA信息
然后安装对应的CUDA(版本可小于等于电脑对应的信息)
英伟达官网下载链接:https://developer.nvidia.com/cuda-toolkit-archive
下载完CUDA之后,要安装与其对应版本的cuDNN(必须对应!!! )
二、Pytorch安装(划重点!!!)
接下来就是进行Pytorch的安装
我推荐是使用的是Anaconda进行pytorch进行安装下载,因为便于管理。
Anaconda详细安装及使用教程(带图文)链接:https://blog.csdn.net/ITLearnHall/article/details/81708148
pytorch安装https://blog.csdn.net/qq_45281807/article/details/121294644
但是!!!安装这种方法可能会一直显示false,因为清华源不支持pytorch了,如果用conda默认下载的,会出现cpu的版本。
这个时候,我们就需要手动下载安装包,进行配置咯~
手动下载配置:
不使用命令,离线下载之后安装即可(根据官网的下载命令查找对应版本号)CUDA,Python,torch,torchvision版本号一定要对!!
各种版本的torch,torchvision,torchaudio下载链接表单https://download.pytorch.org/whl/torch_stable.html
安装上述.whl文件
pip install wheel
pip install 路径\文件名.whl
验证下载Pytorch是否成功
python
import torch
import torchvision
print(torch.cuda.is_available())
返回True表示CUDA配置成功~
参考教程: