概述
驱动 --> anaconda --> pytorch
- 驱动cuda版本越新越好，需要高于pytorch所需的cuda。
- cuda可以结合需要的pytorch版本一起在anaconda内安装。
- 安装不同cuda版本时只需在anaconda内新建环境。新建环境时注意python的版本需要能够支持pytorch。
- cudnn可以在环境内继续安装。
- 最好能够先通外网。
过程
一、驱动
- 卸载低版本nvidia驱动
- 安装新版本驱动
1）官网地址：https://www.nvidia.cn/Download/index.aspx?lang=cn
2）选择对应版本：
image.png
3）下载的驱动版本信息备忘：
image.png
二、安装anaconda
官网地址：https://www.anaconda.com/
版本信息：使用免费的个人版，似乎够用了
Anaconda3-2021.05-Windows-x86_64
image.png
三、安装pytorch
- 安装实验所需的旧环境
pytorch1.3.1 cuda9.2
1）先新建环境，选择python3.7
2）获取安装命令
上pytorch官网首页，进入旧版链接
https://pytorch.org/get-started/previous-versions/
image.png
选择命令，注意是conda下的windows：
image.png
直接修改命令，改成需要的版本：
conda install pytorch==1.3.1 torchvision==0.4.2 cudatoolkit=9.2 -c pytorch
3）安装
进入相应环境的命令行界面：
image.png
使用命令行安装，要外网，非外网自找镜像。
4）安装cudnn
在相应环境下选择未安装组件，从中找到cudnn
image.png
5）安装后版本信息：
image.png
image.png
image.png
-
其他版本pytorch
1）新建环境，选择python3.8
2）命令
image.png
其余同上。
