如何搭建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 相关文章推荐
Django的HttpRequest和HttpResponse对象详解
Jan 26 Python
详解Python中的正则表达式
Jul 08 Python
selenium+python实现1688网站验证码图片的截取功能
Aug 14 Python
python将list转为matrix的方法
Dec 12 Python
Django之模型层多表操作的实现
Jan 08 Python
基于python实现的百度音乐下载器python pyqt改进版(附代码)
Aug 05 Python
selenium 多窗口切换的实现(windows)
Jan 18 Python
python实现飞机大战游戏(pygame版)
Oct 26 Python
Python HTMLTestRunner如何下载生成报告
Sep 04 Python
关于python中remove的一些坑小结
Jan 04 Python
Django rest framework如何自定义用户表
Jun 09 Python
Python接口自动化之文件上传/下载接口详解
Apr 05 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
session 的生命周期是多长
2006/10/09 PHP
JoshChen_web格式编码UTF8-无BOM的小细节分析
2013/08/16 PHP
php实现遍历目录并删除指定文件中指定内容
2015/01/21 PHP
理解php依赖注入和控制反转
2016/05/11 PHP
Yii CDBCriteria常用方法实例小结
2017/01/19 PHP
PHP implode()函数用法讲解
2019/03/08 PHP
javascript实用小函数使用介绍
2013/11/11 Javascript
ParseInt函数参数设置介绍
2014/01/02 Javascript
jq实现酷炫的鼠标经过图片翻滚效果
2014/03/12 Javascript
javascript数据类型示例分享
2015/01/19 Javascript
javascript实现点击按钮让DIV层弹性移动的方法
2015/02/24 Javascript
JavaScript使用concat连接数组的方法
2015/04/06 Javascript
Jquery判断form表单数据是否变化
2016/03/30 Javascript
javascript中活灵活现的Array对象详解
2016/11/30 Javascript
Vue学习之axios的使用方法实例分析
2020/01/06 Javascript
Vue中el-form标签中的自定义el-select下拉框标签功能
2020/04/20 Javascript
代码讲解Python对Windows服务进行监控
2018/02/11 Python
Python下利用BeautifulSoup解析HTML的实现
2020/01/17 Python
PyCharm如何导入python项目的方法
2020/02/06 Python
Python3开发实例之非关系型图数据库Neo4j安装方法及Python3连接操作Neo4j方法实例
2020/03/18 Python
CSS3教程(9):设置RGB颜色
2009/04/02 HTML / CSS
基于html5 DeviceOrientation 实现微信摇一摇功能
2015/09/25 HTML / CSS
linux面试题参考答案(11)
2012/05/01 面试题
土木工程建筑专业毕业生求职信
2013/10/21 职场文书
绩效工资实施方案
2014/03/15 职场文书
篝火晚会策划方案
2014/05/16 职场文书
九一八事变纪念日演讲稿
2014/09/14 职场文书
给上级领导的感谢信
2015/01/22 职场文书
幼儿园欢迎词范文
2015/01/26 职场文书
教师年度考核个人总结
2015/02/12 职场文书
2015社区六五普法工作总结
2015/04/21 职场文书
银行安全保卫工作总结
2015/08/10 职场文书
原生JS实现飞机大战小游戏
2021/06/09 Javascript
Nginx反向代理配置的全过程记录
2021/06/22 Servers
Python集合set()使用的方法详解
2022/03/18 Python
Spring Boot接口定义和全局异常统一处理
2022/04/20 Java/Android