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编辑器里面的效果: