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 安装,选择自定义安装
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第1张图片
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第2张图片
1.2.3 安装位置默认(不会太占用内存),如诺不然,需配置环境变量PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第3张图片1.2.4 安装后,和英伟达cuda相关的程序如下图所示。
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第4张图片
注意,千万不要勾选 Nsight Visual Studio Edition 2019.2等类似的无用的东西。

1.3 验证CUDA是否已安装
运行cmd,输入nvcc --version 即可查看版本号;
set cuda,可以查看cuda设置的环境变量。
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第5张图片
2. python3.6.6 下载安装:(pip)
2.1 python3.6.6下载
网址:https://www.python.org/downloads/release/python-366/
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第6张图片
在这里插入图片描述
2.2 安装
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第7张图片
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第8张图片
注释:这里选择全安装,后期需要使用pip安装PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第9张图片

PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第10张图片
2.3 验证是否安装成功,使用系统的cmd命令:
在这里插入图片描述
3. jupyter notebook安装
3.1 首先查看系统的pip
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第11张图片
3.2 采用指令pip install jupyter notebook 安装即可
3.3 运行jupyter
在终端输入jupyter notebook后,会弹出如下页面
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第12张图片
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第13张图片
3.4 修改Jupyter Notebook文件默认保存路径(默认保存在C盘下,后期需要下载数据库,还是更改目录比较合理)
3.4.1 生成默认配置文件
运行cmd, 执行 jupyter notebook --generate-config

3.4.2 编辑配置文件
将 C:\Users\xiyou.jupyter\jupyter_notebook_config.py 中:
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第14张图片
去掉注释,并修该为相应目标路径(路径必须存在)

3.4.3 重启Jupyter Notebook
打印当前路径:
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第15张图片
4.小结
本文进行了PyTorch环境的搭建,win10下,使用jupyter notebook编程,python版本python3.6.6,采用cuda10.1进行加速。
下面看看在jupyter编辑器里面的效果:
PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)_第16张图片

你可能感兴趣的:(PyTorch)