如何搭建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 pass 语句使用示例
Mar 11 Python
python3编码问题汇总
Sep 06 Python
Python爬虫之正则表达式基本用法实例分析
Aug 08 Python
在django中,关于session的通用设置方法
Aug 06 Python
python sorted函数的小练习及解答
Sep 18 Python
django 实现celery动态设置周期任务执行时间
Nov 19 Python
python2 对excel表格操作完整示例
Feb 23 Python
Python实现初始化不同的变量类型为空值
Jun 02 Python
新版Pycharm中Matplotlib不会弹出独立的显示窗口的问题
Jun 02 Python
Python实现爬取并分析电商评论
Jun 19 Python
mac安装python3后使用pip和pip3的区别说明
Sep 01 Python
写一个Python脚本下载哔哩哔哩舞蹈区的所有视频
May 31 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中文汉字验证码
2007/04/08 PHP
完美利用Yii2微信后台开发的系列总结
2016/07/18 PHP
laravel 解决paginate查询多个字段报错的问题
2019/10/22 PHP
几个常用的JavaScript字符串处理函数 - split()、join()、substring()和indexOf()
2009/06/02 Javascript
JS 学习笔记 防止发生命名冲突
2009/07/30 Javascript
JavaScript 类型的包装对象(Typed Wrappers)
2011/10/27 Javascript
php+ajax+jquery实现点击加载更多内容
2015/05/03 Javascript
JS获取元素多层嵌套思路详解
2016/05/16 Javascript
浅谈javascript:两种注释,声明变量,定义函数
2016/09/29 Javascript
JavaScript中Math对象的方法介绍
2017/01/05 Javascript
JavaScript之浏览器对象_动力节点Java学院整理
2017/07/03 Javascript
js es6系列教程 - 基于new.target属性与es5改造es6的类语法
2017/09/02 Javascript
AngularJS实现注册表单验证功能
2017/10/16 Javascript
微信小程序实现添加手机联系人功能示例
2017/11/30 Javascript
vue 组件之间事件触发($emit)与event Bus($on)的用法说明
2020/07/28 Javascript
vue3.0 上手体验
2020/09/21 Javascript
Python实现约瑟夫环问题的方法
2016/05/03 Python
python 全局变量的import机制介绍
2017/09/07 Python
python+matplotlib绘制旋转椭圆实例代码
2018/01/12 Python
Python中按键来获取指定的值
2019/03/02 Python
详解python中list的使用
2019/03/15 Python
python语言线程标准库threading.local解读总结
2019/11/10 Python
python MultipartEncoder传输zip文件实例
2020/04/07 Python
浅谈在JupyterNotebook下导入自己的模块的问题
2020/04/16 Python
使用anaconda安装pytorch的实现步骤
2020/09/03 Python
波兰补充商店:Muscle Power
2018/10/29 全球购物
印度排名第一的蛋糕、鲜花和礼品送货:Winni
2019/08/02 全球购物
迟到检讨书1000字
2014/01/15 职场文书
借名购房协议书范本
2014/10/06 职场文书
延安红色之旅心得体会
2014/10/07 职场文书
单位政审意见范文
2015/06/04 职场文书
红十字会救护培训简讯
2015/07/20 职场文书
销售口号霸气押韵
2015/12/24 职场文书
vue实现滑动解锁功能
2022/03/03 Vue.js
服务器间如何实现文件共享
2022/05/20 Servers