基于全新电脑环境安装pytorch的GPU版本

基于全新电脑环境安装pytorch的GPU版本

前言：

距离第一次安装深度学习的GPU环境已经过去了4年多（当时TensorFlow特别麻烦），现在发现安装pytorch的GPU版本还是很简单方便的，流程记录如下。

安装步骤：

步骤一：官网下载Anaconda

Free Download | Anaconda

直接下载最新版本到电脑里，并安装。

步骤二：查询电脑的CUDA Version

win+R 然后输入cmd调出命令窗，输入 nvidia-smi

步骤三：确定电脑GPU的NVDIA型号。

通过搜索找到”设备管理器”，再找到其中的“显示适配器”。

步骤四：更新NVIDIA驱动程序

NVIDIA官网：/

 选择对应的硬件环境

 搜索到合适的驱动后，下载并安装（可以用C盘的默认位置）。

 安装完毕后，重启电脑。

步骤五：再次查询CUDA Version

win+R 然后输入cmd调出命令窗，输入 nvidia-smi

可以看到，CUDA Version的版本已经从11.7变成了12.2。

步骤六：通过官网下载pytorch

官网地址：/

注意：CUDA Version要选择比电脑低的情况，我这里选择了默认推荐版本。

在Anaconda的prompt中输入官网指令。

 pip3 install torch torchvision torchaudio --index-url

在安装过程中，遇到的选择，全部选择y。

其中，由于没有借助镜像源（增加出问题的概率），安装过程较慢（大概花了30分钟）。

 步骤七：在anaconda的prompt环境测试安装结果

显示True，安装成功。

步骤八：打开anaconda自带的spyder编辑器进行测试

步骤九：跑一个MNIST数据集分类样例代码测试

 nvidia-smi -l 1实时观察GPU的占用率

 补充说明

（1）windows系统需要增加Anaconda的环境变量。 

（2）matplotlib通常需要重新安装对应版本，很容易版本不匹配。