如何搭建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 相关文章推荐
TensorFlow 模型载入方法汇总(小结)
Jun 19 Python
Django forms组件的使用教程
Oct 08 Python
Python3.6.2调用ffmpeg的方法
Jan 10 Python
Python面向对象程序设计构造函数和析构函数用法分析
Apr 12 Python
Python使用Beautiful Soup爬取豆瓣音乐排行榜过程解析
Aug 15 Python
Python 中list ,set,dict的大规模查找效率对比详解
Oct 11 Python
pycharm第三方库安装失败的问题及解决经验分享
May 09 Python
python中类与对象之间的关系详解
Dec 16 Python
详解Python中的Lock和Rlock
Jan 26 Python
Python基础知识之变量的详解
Apr 14 Python
opencv 分类白天与夜景视频的方法
Jun 05 Python
Python使用华为API为图像设置多个锚点标签
Apr 12 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
一个从别的网站抓取信息的例子(域名查询)
2006/10/09 PHP
php 无法加载mysql的module的时候的配置的解决方案引发的思考
2012/01/27 PHP
php中并发读写文件冲突的解决方案
2013/10/25 PHP
php计算几分钟前、几小时前、几天前的几个函数、类分享
2014/04/09 PHP
jquery选择器-根据多个属性选择示例代码
2013/10/21 Javascript
js实现浏览器的各种菜单命令比如打印、查看源文件等等
2013/10/24 Javascript
jquery ajax应用中iframe自适应高度问题解决方法
2014/04/12 Javascript
淘宝网提供的国内NPM镜像简介和使用方法
2014/04/17 Javascript
ExtJS4给Combobox设置列表中的默认值示例
2014/05/02 Javascript
Nodejs全栈框架StrongLoop推荐
2014/11/09 NodeJs
Javascript基础教程之JavaScript语法
2015/01/18 Javascript
JavaScript编写页面半透明遮罩效果的简单示例
2016/05/09 Javascript
JS实现焦点图轮播效果的方法详解
2016/12/19 Javascript
JavaScript实现鼠标滚轮控制页面图片切换功能示例
2017/10/14 Javascript
微信小程序页面跳转功能之从列表的item项跳转到下一个页面的方法
2017/11/27 Javascript
bootstrap中selectpicker下拉框使用方法实例
2018/03/22 Javascript
vue中的适配px2rem示例代码
2018/11/19 Javascript
layui原生表单验证的实例
2019/09/09 Javascript
基于layui轮播图满屏是高度自适应的解决方法
2019/09/16 Javascript
Python 2.7.x 和 3.x 版本的重要区别小结
2014/11/28 Python
Python爬虫中urllib库的进阶学习
2018/01/05 Python
详解重置Django migration的常见方式
2019/02/15 Python
django-allauth入门学习和使用详解
2019/07/03 Python
python将类似json的数据存储到MySQL中的实例
2019/07/12 Python
Python3使用xml.dom.minidom和xml.etree模块儿解析xml文件封装函数的方法
2019/09/23 Python
对pytorch的函数中的group参数的作用介绍
2020/02/18 Python
CSS3自定义滚动条样式的示例代码
2017/08/21 HTML / CSS
HTML5梦幻之旅——炫丽的流星雨效果实现过程
2013/08/06 HTML / CSS
工业设计毕业生自荐信
2014/04/13 职场文书
高职教师先进事迹材料
2014/08/24 职场文书
工会文体活动总结
2015/05/07 职场文书
纪检监察立案决定书
2015/06/24 职场文书
2016年九九重阳节活动总结
2016/04/01 职场文书
送给小学生的暑假礼物!小学生必背99首古诗
2019/07/02 职场文书
详解Nginx启动失败的几种错误处理
2021/04/01 Servers
Win11 Build 22000.51版本文件资源管理器“命令栏”和上下文菜单有什么新变化?
2021/11/21 数码科技