如何搭建pytorch环境的方法步骤


Posted in Python onMay 06, 2020

1.conda创建虚拟环境pytorch_gpu

conda create -n pytorch_gpu python=3.6

创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无需很长,成功之后界面如下所示:

如何搭建pytorch环境的方法步骤

2.切换到pytorch环境

使用如下命令,切换到我们刚刚创建好的pytorch虚拟环境,这样我们避免与其它python环境之间的干扰。

conda activeta pytorch_gpu

切换成功之后就会看到在路径前边显示我们已经进入该虚拟环境。

如何搭建pytorch环境的方法步骤

3.安装几个常用库(也可暂时不安)

conda install pandas jupyter notebook

4.安装pytorch

4.1进入官网查看要下载的版本

查看对应的版本,这里是官方链接:

如何搭建pytorch环境的方法步骤

4.2 根据系 统信息及cuda版本选择对应toolkit

这里最主要的是那个CUDA的版本,此处我选择的是10.1,是因为我的电脑的 cuda版本信息就是这样的。具体的查看方法可在4.5节查阅。

4.3复制上图中最后一行代码到pytorch环境终端

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

这里是下载过程截图:

如何搭建pytorch环境的方法步骤

但是在下载过程中torchvision以及pytorch没有下载成功,

如何搭建pytorch环境的方法步骤

因为这是去其官网下载,下载速度很慢,在上图中我们也可以看出是因为网络错误,网络上也有几种其他的方法,此处我没去验证,我还是让电脑重新下载的,等待时间挺长的,但是因为是在晚上下载的,一早起来就好了

如何搭建pytorch环境的方法步骤

4.4 验证pytorch是否安装成功

此时直接输入 python,即可成功进入:

如何搭建pytorch环境的方法步骤

而后输入如下指令,查看torch是否安装成功

>>> import torch
>>> x=torch.randn(4,4)
>>> print(x)

正常情况下是出现这个界面的:

如何搭建pytorch环境的方法步骤

验证完成之后,可以quit()保存退出。

4.5 如何查看自己电脑cuda版本

4.5.1 windows如何查看

NVDIA控制面板?>帮助?>系统信息

如何搭建pytorch环境的方法步骤

组件?>NVCUDA.DLL 可以查看CUDA版本

如何搭建pytorch环境的方法步骤

这里我的显示是10.1,所以我上边下载的版本也是10.1的,这里的版本要对应上,否则会出现问题。

4.5.2 linux如何查看

打开终端,输入:nvcc -V

nvcc -V

或者如下方式查看:

CUDA:

cat /usr/local/cuda/version.txt

cudnn:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

到此这篇关于如何搭建pytorch环境的方法步骤的文章就介绍到这了,更多相关pytorch搭建环境内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
python打开网页和暂停实例
Sep 30 Python
Python2.x版本中基本的中文编码问题解决
Oct 12 Python
python实现输入数字的连续加减方法
Jun 22 Python
python 3调用百度OCR API实现剪贴板文字识别
Sep 04 Python
对Python闭包与延迟绑定的方法详解
Jan 07 Python
django-rest-swagger的优化使用方法
Aug 29 Python
Python实现socket非阻塞通讯功能示例
Nov 06 Python
Python彻底删除文件夹及其子文件方式
Dec 23 Python
python实现快递价格查询系统
Mar 03 Python
Python startswith()和endswith() 方法原理解析
Apr 28 Python
python实现canny边缘检测
Sep 14 Python
使用python向MongoDB插入时间字段的操作
May 18 Python
Python基于Tkinter编写crc校验工具
May 06 #Python
使用Python实现批量ping操作方法
May 06 #Python
Python super()函数使用及多重继承
May 06 #Python
python中shell执行知识点
May 06 #Python
Python 通过监听端口实现唯一脚本运行方式
May 05 #Python
python2.7使用scapy发送syn实例
May 05 #Python
python 使用raw socket进行TCP SYN扫描实例
May 05 #Python
You might like
DSP接收机前端设想
2021/03/02 无线电
ThinkPHP连接数据库的方式汇总
2014/12/05 PHP
PHP框架自动加载类文件原理详解
2017/06/06 PHP
XHTML下,JS浮动代码失效的问题
2009/11/12 Javascript
网页中CDATA标记的说明
2010/09/12 Javascript
jquery文本框中的事件应用以输入邮箱为例
2014/05/06 Javascript
详解JavaScript中的异常处理方法
2015/06/16 Javascript
javascript实现禁止鼠标滚轮事件
2015/07/24 Javascript
js中 javascript:void(0) 用法详解
2015/08/11 Javascript
JS实现的手机端精简幻灯片效果
2016/09/05 Javascript
三种方式实现瀑布流布局
2017/02/10 Javascript
浅谈sass在vue注意的地方
2017/08/10 Javascript
关于JavaScript的单双引号嵌套问题
2017/08/20 Javascript
使用JS动态显示文本
2017/09/09 Javascript
Angular利用trackBy提升性能的方法
2018/01/26 Javascript
详解Vue+ElementUI从零开始搭建自己的网站(一、环境搭建)
2019/04/30 Javascript
详解如何在JS代码中消灭for循环
2019/12/11 Javascript
Node.js API详解之 querystring用法实例分析
2020/04/29 Javascript
three.js 如何制作魔方
2020/07/31 Javascript
JS数组转字符串实现方法解析
2020/09/04 Javascript
nuxt.js服务端渲染中axios和proxy代理的配置操作
2020/11/06 Javascript
零基础写python爬虫之抓取百度贴吧代码分享
2014/11/06 Python
利用Python的Flask框架来构建一个简单的数字商品支付解决方案
2015/03/31 Python
Python实现括号匹配方法详解
2020/02/10 Python
python3通过qq邮箱发送邮件以及附件
2020/05/20 Python
Python pandas如何向excel添加数据
2020/05/22 Python
阳光体育:Sunny Sports(购买露营和远足设备)
2018/08/07 全球购物
Godiva巧克力英国官网:比利时歌帝梵巧克力
2018/08/28 全球购物
美国在线打印网站:Overnight Prints
2018/10/11 全球购物
J2EE相关知识面试题
2013/08/26 面试题
外贸英语毕业生自荐信
2013/11/14 职场文书
合同协议书格式
2014/04/18 职场文书
爱耳日活动总结
2014/04/30 职场文书
优秀管理者事迹材料
2014/05/22 职场文书
党委书记群众路线对照检查材料思想汇报
2014/10/04 职场文书
入党积极分子培养联系人意见
2015/08/12 职场文书