为了Pytorch的安装和GPU运算可让人花了不少力气。
今天说一说CUDA + cuDNN安装步骤:
(1)驱动下载(这里放的是适配我电脑的,找一个适配电脑型号的):https://www.nvidia.com/download/driverResults.aspx/153944
(2)CUDA toolkit下载,在官网下载(需要注册):https://developer.nvidia.com/cuda-toolkit
(3)cuDNN下载:https://developer.nvidia.com/cudnn
这三个步骤完成后,所有的下载工作算是完成了!
接下来就是安装:(1)和(2)我建议无脑安装也不要换什么文件位置。(3)是插入式安装,不能直接下一步,在(1)和(2)安装完成之后,将(3)下载的cuDNN解压,然后按照这位大哥的思路走完即可:https://zhuanlan.zhihu.com/p/94220564
当这一切都安装完的时候,开始GPU计算,发现GPU不够用才是硬伤/(ㄒoㄒ)/~~