pytorch安装及其版本切换

一、安装说明

pytorch是动态库，依赖的库各个版本的pytorch可能是不同的，所以pytorch和依赖库有版本对应关系。如下：
https://cloud.tencent.com/developer/article/2038557
安装pytorch需要提前安装对应的cuda本版，具体版本查询上面的链接。（安装cuda时要注意driver version的要求，不然不能运行。查看driver version可以在NVIDIA的控制面板查看）。

二、安装步骤

这里我需要下载安装pytorch 1.4

1. 下载安装cuda

（1）查看版本

1.4版本的pytorch需要10.1的cuda驱动。

（2）下载

cuda历代版本下载

选择对应版本进入下载页面（推荐local下载，如果安装失败不需要重新下载，节省流量和时间）

（3）安装

安装完成(这里我以前安装了10.0的版本，所以有个v10.0)

（4）环境变量配置

安装完默认添加CUDA_PATH和CUDA PATH_V10_1的环境变量

在path环境变量中也有添加
到此cuda安装完成

2. pytorch的安装

pytorch的安装较为简单，可以直接用命令行安装
用conda新建一个虚拟环境便于pytorch版本的管理

conda create -n pytorch14 python=3.7

激活环境

activate pytorch14

安装pytorch
命令行在官网进行查询
pytorch安装命令行

输入命令行即可安装

报错处理

安装出现错误，可以自行下载whl文件，通过pip安装。
参考https://blog.csdn.net/candice5566/article/details/104929677
https://blog.csdn.net/weixin_43332715/article/details/123518061
下载地址

验证

输出True说明安装成功

三、本版切换

因为pytorch和cuda是版本对应的，所以切换pytorch版本需要切换对应的cuda版本。

1. 切换cuda环境变量

2. 切换pytorch对应的虚拟环境