Win10—pytorch环境配置

 

本文中的配置操作来源于沐神的视频教程，网址如下：

Windows 下安装 CUDA 和 Pytorch 跑深度学习 - 动手学深度学习v2_哔哩哔哩 (゜-゜)つロ 干杯~-bilibili

 

pytorch的GPU环境配置真的是相当简单啊，想之前受tf环境配置折磨的日子，简直要发疯。接下来是具体操作流程：

1. 查看显卡信息。

在cmd框框中输入dxdiag 即可。

注：貌似只支持Nvidia 的独显。

1. 进入cuda官网进行下载，各项选择如下图，最后点击Download，安装时按照默认安装。

 

安装完之后，可进行命令行窗口，输入nvidia-smi，查看cuda是否成功安装，即查看cuda版本信息。

若 输入nvidia-smi 出现报错，提示 不是内部或外部命令，也不是可运行的程序。

则 将以下路径添加到环境变量中即可。具体内容可查看以下博主博客。

C:\Program Files\NVIDIA Corporation\NVSMI

 

来自

 

注1：这里下载的是11.3版本，视频中是11.2.沐神在视频中说道：小版本的cuda更新迭代应该影响不大。大版本的变换就不行，如10和11。

注2：关于环境变量，这里本文中未进行探索。由于博主原来的笔记本中安装了cuda10，所以直接将原cuda10的环境变量改成了11.3。为保证安装顺利，还是建立配置下相关环境变量，具体操作网上冲浪即可。

注3：这里不需要再额外下载cudnn，具体原因未知，可能是下面下载pytorch时会一块安装。

 

 

1. 进入miniconda官网，进行下载

下载下图中的Python 3.8版本，目前一般电脑都是64位的，选择64-bit。安装时按照默认安装就好。

安装好之后，可进入miniconda的命令行窗口，输入 python --version  查看python版本，即检查是否安装成功。

 

 

 

1. 进入pytorch官网，点击install，选择如下图所示

再进入miniconda命令行窗口，输入 pip3…..等命令

注：请一定注意是在miniconda的命令行窗口，而不是系统的，博主在这里栽了几回。由于是在外网官网下载，所以用国内网下载很慢，经常会出现read time out的报错提示。

所以建议有条件的尽量使用校园网，实在不行就只能寻找镜像下载了。另外，一次下载大概是3个g左右，因此用校园网的朋友注意不要下错地方。

 

 

1. 安装完毕之后，进入miniconda命令行窗口，输入以下命令，若出现如图的情况，就表明安装成功了！！！

 

 

以上为安装过程，如有错误，欢迎指出！