PyTorch环境搭建（window10 + cuda10.1 + python3.6 + pip + jupyter notebook）

PyTorch环境搭建（window10 + cuda10.1 + python3.6 + pip + jupyter notebook）

所有的伟大源于一个勇敢的开始

1. cuda10.1下载安装：
1.1 cuda下载
网址：添加链接描述
1.2 安装
1.2.1 离线文档：

1.2.2 安装，选择自定义安装


1.2.3 安装位置默认（不会太占用内存），如诺不然，需配置环境变量1.2.4 安装后，和英伟达cuda相关的程序如下图所示。

注意，千万不要勾选 Nsight Visual Studio Edition 2019.2等类似的无用的东西。

1.3 验证CUDA是否已安装
运行cmd，输入nvcc --version 即可查看版本号；
set cuda，可以查看cuda设置的环境变量。

2. python3.6.6 下载安装：（pip）
2.1 python3.6.6下载
网址：https://www.python.org/downloads/release/python-366/


2.2 安装


注释：这里选择全安装，后期需要使用pip安装

2.3 验证是否安装成功，使用系统的cmd命令：

3. jupyter notebook安装
3.1 首先查看系统的pip

3.2 采用指令pip install jupyter notebook 安装即可
3.3 运行jupyter
在终端输入jupyter notebook后，会弹出如下页面


4.小结
本文进行了PyTorch环境的搭建，win10下，使用jupyter notebook编程，python版本python3.6.6，采用cuda10.1进行加速。
下面看看在jupyter编辑器里面的效果：