如何搭建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的web框架中的Model的教程
Apr 29 Python
Python中使用bidict模块双向字典结构的奇技淫巧
Jul 12 Python
Linux RedHat下安装Python2.7开发环境
May 20 Python
python安装pywin32clipboard的操作方法
Jan 24 Python
python实现简单日期工具类
Apr 24 Python
python wxpython 实现界面跳转功能
Dec 17 Python
Python3 全自动更新已安装的模块实现
Jan 06 Python
python模拟实现斗地主发牌
Jan 07 Python
python默认参数调用方法解析
Feb 09 Python
pytorch:model.train和model.eval用法及区别详解
Feb 20 Python
python实现图片转字符画的完整代码
Feb 21 Python
自动在Windows中运行Python脚本并定时触发功能实现
Sep 04 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
PHP中通过fopen()函数访问远程文件示例
2014/11/18 PHP
PHP CURL post数据报错 failed creating formpost data
2016/10/16 PHP
浅析PHP7的多进程及实例源码
2019/04/14 PHP
PHP容器类的两种实现方式示例
2019/07/24 PHP
25个好玩的JavaScript小游戏分享
2011/04/22 Javascript
js读取被点击次数的简单实例(从数据库中读取)
2014/03/07 Javascript
JavaScript两种跨域技术全面介绍
2014/04/16 Javascript
jquery实现的图片点击滚动效果
2014/04/29 Javascript
js获取json元素数量的方法
2015/01/27 Javascript
js实现星星打分效果的方法
2020/07/05 Javascript
javascript完美实现给定日期返回上月日期的方法
2017/06/15 Javascript
vue 文件目录结构详解
2017/11/24 Javascript
bootstrap3中container与container_fluid外层容器的区别讲解
2017/12/04 Javascript
jQuery实现文件编码成base64并通过AJAX上传的方法
2018/04/12 jQuery
Vue2.0实现调用摄像头进行拍照功能 exif.js实现图片上传功能
2018/04/28 Javascript
vue-cli+axios实现文件上传下载功能(下载接收后台返回文件流)
2019/05/10 Javascript
深入了解响应式React Native Echarts组件
2019/05/29 Javascript
解决在Vue中使用axios用form表单出现的问题
2019/10/30 Javascript
vue router 传参获取不到的解决方式
2019/11/13 Javascript
javscript 数组扁平化的实现
2020/02/03 Javascript
Vuex模块化应用实践示例
2020/02/03 Javascript
[03:40]DOTA2亚洲邀请赛小组赛第二日 赛事回顾
2015/01/31 DOTA
python类和继承用法实例
2015/07/07 Python
解决python matplotlib imshow无法显示的问题
2018/05/24 Python
Django Python 获取请求头信息Content-Range的方法
2019/08/06 Python
python安装本地whl的实例步骤
2019/10/12 Python
手工制作的豪华英式沙发和沙发床:Willow & Hall
2019/05/03 全球购物
香港艺人陈冠希创办的潮流品牌:JUICESTORE
2021/03/04 全球购物
大学生应聘求职信
2014/05/26 职场文书
水污染治理工程专业自荐信
2014/06/21 职场文书
助残日活动总结
2014/08/27 职场文书
安全员岗位职责
2015/02/10 职场文书
医药公司采购员岗位职责
2015/04/03 职场文书
2015最新婚礼司仪主持词
2015/06/30 职场文书
详解Laravel服务容器的优势
2021/05/29 PHP
排查并解决MySQL生产库内存使用率高的报警
2022/04/11 MySQL