最近因为跑transformer模型,要配置环境。torch_cpu版本比较好处理,直接在anaconda上输入对应的命令行就可以自动下载。比较经典的事torch1.8.1,在很多教程和案例中都使用到了这个torch。
不经常使用N卡,所以驱动、cuda、cudnn也要重新安装。
安装驱动没什么讲究,直接通过网址选择对应的驱动下载安装即可:NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA
顺便附上torch的下载地址,里面包含了多种torch_cuda版本:download.pytorch.org/whl/torch_stable.html
下载cuda要和torch_cuda的版本号对应,比如torch_cuda11.1要想在对应的11.1CUDA:
CUDA Toolkit Archive | NVIDIA Developer
重点:cuda安装的时候,exe文件解压的位置不要和后面软件安装的位置一致,不然软件安装完成后,解压文件夹的内容会别删除。这也是我疑惑了很久的问题,为什么每次安装完成都没有。
安装CUDA完成之前需要配置系统变量,变量是什么安装过程有提示的。
最后是cudnn的安装,其实就是个压缩包解压,把对应文件夹的内容放到cuda里面:cuDNN Archive | NVIDIA Developer