主要流程分4步
安装驱动→安装cuda→安装cudann→安装pytorch
step1 安装驱动
在官网下载最新版本,这个驱动是最新的可以支持所有版本,所以盲目直接下官方最新的就可以了
链接如下:
https://www.nvidia.cn/geforce/geforce-experience/
按照提示一直安装就行
step2 安装cuda 就是安装cuda toolkit
这个需要根据自己的显卡版本决定,如果是3060Ti算力是8.6,对于的就是11.1-11.4版本的cuda toolkit
具体可以看csdn这个博主的博文(注:侵权删):
https://blog.csdn.net/weixin_43667077/article/details/135654738
所以直接到官方网站如下:
https://developer.nvidia.com/cuda-toolkit-archive
注意,我一开始就直接选了11.4,但是去安装pytorch的时候发现只有11.3对应的,如果要一一对应的就选11.3,稳妥些,虽然我现在装的11.4的cuda和11.3的pytorch也一样能用,但不知道后续会不会出问题,看网上很多都是选对应的cuda和pytorch
step3 安装cudann (需要注册登录)
这个也很简单,就是在官网下载对应版本的cudann就行,我们这里是11.4所以就下载11的cudann,网址和图片如下:
https://developer.nvidia.com/cudnn-downloads
我是直接点击最新的本版cudann9.1.1再选cuda11的版本,其他的版本可以按下面的链接,因为我的也运行成功了,所以cudann版本新也没事,后面选配置的时候cuda选择对应的11就行。
https://developer.nvidia.com/cudnn-archive
下载好以后,解压,然后把解压后的文件中的三个文件夹复制,粘贴到刚刚cuda toolkit安装的位置,覆盖替换
(如果找不到的话,可以win+R,输入cmd,然后弹出来的窗口输入 set cuda
就可以看到cuda toolkit安装的位置了
然后这一步就结束了
step4 安装pytorch,这个直接在官网选好配置安装就好
pytorch网址如下:
https://pytorch.org/get-started/locally/
配置根据自己的cuda toolkit选择如下,然后复制pip那部分到anaconda prompt运行:
测试
在anaconda prompt 输入python进入python环境
然后输入
import torch
torch.cuda.is_available()
可以看到返回
可以参考:
https://mp.weixin.qq.com/s/M-zm2VbgAl4zzyejIqTxHA
https://mp.weixin.qq.com/s/MjhmCTAnNObJCCpIEPoXgA
#小程序://小红书/Cydt1hLBvPZvP7I