如何搭建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实现文本去重且不打乱原本顺序
Jan 26 Python
详解Python pygame安装过程笔记
Jun 05 Python
python 创建弹出式菜单的实现代码
Jul 11 Python
利用Pycharm断点调试Python程序的方法
Nov 29 Python
Python从文件中读取数据的方法讲解
Feb 14 Python
Python使用统计函数绘制简单图形实例代码
May 15 Python
python的set处理二维数组转一维数组的方法示例
May 31 Python
Tensorflow 自定义loss的情况下初始化部分变量方式
Jan 06 Python
浅谈pytorch中torch.max和F.softmax函数的维度解释
Jun 28 Python
Python 多线程之threading 模块的使用
Apr 14 Python
Python下opencv库的安装过程及问题汇总
Jun 11 Python
pd.drop_duplicates删除重复行的方法实现
Jun 16 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
ThinkPHP php 框架学习笔记
2009/10/30 PHP
PHP扩展模块memcached长连接使用方法分析
2014/12/24 PHP
php中namespace use用法实例分析
2016/01/22 PHP
php实现将数据做成json的格式给前端使用
2018/08/21 PHP
List the Codec Files on a Computer
2007/06/11 Javascript
jQuery AjaxQueue改进步骤
2011/10/06 Javascript
浅谈Javascript Base64 加密解密
2014/12/28 Javascript
js用拖动滑块来控制图片大小的方法
2015/02/27 Javascript
js图片卷帘门导航菜单特效代码分享
2015/09/10 Javascript
JS事件添加和移出的兼容写法示例
2016/06/20 Javascript
js HTML5多图片上传及预览实例解析(不含前端的文件分割)
2016/08/26 Javascript
纯JavaScript 实现flappy bird小游戏实例代码
2016/09/27 Javascript
AngularJS入门教程之过滤器用法示例
2016/11/02 Javascript
微信小程序 POST请求的实例详解
2017/09/29 Javascript
Vue v2.4中新增的$attrs及$listeners属性使用教程
2018/01/08 Javascript
浅析node应用的timing-attack安全漏洞
2018/02/28 Javascript
vue路由拦截及页面跳转的设置方法
2018/05/24 Javascript
JS实现区分中英文并统计字符个数的方法示例
2018/06/09 Javascript
antd多选下拉框一行展示的实现方式
2020/10/31 Javascript
浅谈Vue开发人员的7个最好的VSCode扩展
2021/01/20 Vue.js
Python 序列的方法总结
2016/10/18 Python
Python pyinotify日志监控系统处理日志的方法
2018/03/08 Python
利用Python批量提取Win10锁屏壁纸实战教程
2018/03/27 Python
python如何爬取个性签名
2018/06/19 Python
关于python写入文件自动换行的问题
2018/06/23 Python
Python使用gRPC传输协议教程
2018/10/16 Python
Flask框架工厂函数用法实例分析
2019/05/25 Python
使用python批量修改文件名的方法(视频合并时)
2020/03/24 Python
CSS3中的常用选择器使用示例整理
2016/06/13 HTML / CSS
琳达·法罗眼镜英国官网:Linda Farrow英国
2021/01/19 全球购物
简述网络文件系统NFS,并说明其作用
2016/10/19 面试题
法律进机关实施方案
2014/03/12 职场文书
乔丹名人堂演讲稿
2014/05/24 职场文书
小学秋季运动会报道稿
2014/09/30 职场文书
导游词之山东孔庙
2019/11/04 职场文书
python实现简单的名片管理系统
2021/04/26 Python