您的位置:首页 > 编程语言 > Python开发

PyTorch环境搭建(window10 + cuda10.1 + python3.6 + pip + jupyter notebook)

2020-07-16 11:18 176 查看

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后,会弹出如下页面


3.4 修改Jupyter Notebook文件默认保存路径(默认保存在C盘下,后期需要下载数据库,还是更改目录比较合理)
3.4.1 生成默认配置文件
运行cmd, 执行 jupyter notebook --generate-config

3.4.2 编辑配置文件
将 C:\Users\xiyou.jupyter\jupyter_notebook_config.py 中:

去掉注释,并修该为相应目标路径(路径必须存在)

3.4.3 重启Jupyter Notebook
打印当前路径:

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

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: