如何搭建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获取Linux系统的各种信息
Jul 10 Python
Python3实现将文件树中所有文件和子目录归档到tar压缩文件的方法
May 22 Python
Python在图片中添加文字的两种方法
Apr 29 Python
利用Anaconda简单安装scrapy框架的方法
Jun 13 Python
通过python爬虫赚钱的方法
Jan 29 Python
Python一个简单的通信程序(客户端 服务器)
Mar 06 Python
pyqt5 QProgressBar清空进度条的实例
Jun 21 Python
Python通过VGG16模型实现图像风格转换操作详解
Jan 16 Python
python 实现两个npy档案合并
Jul 01 Python
Pycharm 2020.1 版配置优化的详细教程
Aug 07 Python
pycharm中如何自定义设置通过“ctrl+滚轮”进行放大和缩小实现方法
Sep 16 Python
PYTHON 使用 Pandas 删除某列指定值所在的行
Apr 28 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 COOKIE设置为浏览器进程
2009/06/21 PHP
PHP Curl出现403错误的解决办法
2014/05/29 PHP
PHP面试常用算法(推荐)
2016/07/22 PHP
JavaScript面向对象编程
2008/03/02 Javascript
通过event对象的fromElement属性解决热区设置主实体的一个bug
2008/12/22 Javascript
Document 对象的常用方法
2009/07/31 Javascript
Javascript创建自定义对象 创建Object实例添加属性和方法
2012/06/04 Javascript
一个简单的瀑布流效果(主体形式自写)
2013/05/27 Javascript
JS实现静止元素自动移动示例
2014/04/14 Javascript
JS简单实现无缝滚动效果实例
2016/08/24 Javascript
PHP抓取HTTPS内容和错误处理的方法
2016/09/30 Javascript
简单谈谈axios中的get,post方法
2017/06/25 Javascript
vue短信验证性能优化如何写入localstorage中
2018/04/25 Javascript
jQuery实现的滑块滑动导航效果示例
2018/06/04 jQuery
Bootstrap table表格初始化表格数据的方法
2018/07/25 Javascript
javascript数组的定义及操作实例
2019/11/10 Javascript
借助云开发实现小程序短信验证码的发送
2020/01/06 Javascript
[46:09]2014 DOTA2华西杯精英邀请赛 5 25 LGD VS VG第三场
2014/05/26 DOTA
基于Python3.6+splinter实现自动抢火车票
2018/09/25 Python
python之验证码生成(gvcode与captcha)
2019/01/02 Python
Django实现web端tailf日志文件功能及实例详解
2019/07/28 Python
python实现人机猜拳小游戏
2020/02/03 Python
详解python中GPU版本的opencv常用方法介绍
2020/07/24 Python
pycharm永久激活超详细教程
2020/10/29 Python
Python 找出英文单词列表(list)中最长单词链
2020/12/14 Python
理肤泉英国官网:La Roche-Posay英国
2019/01/14 全球购物
高三英语教学反思
2014/01/13 职场文书
班级入场式解说词
2014/02/01 职场文书
还款承诺书范文
2014/05/20 职场文书
小学教师师德师风个人整改措施
2014/09/18 职场文书
乡镇民主生活会发言材料
2014/10/20 职场文书
2015年绩效考核工作总结
2015/05/23 职场文书
单位收入证明范本
2015/06/18 职场文书
家长会后的感想
2015/08/11 职场文书
新员工入职感言范文!
2019/07/04 职场文书
分析mysql中一条SQL查询语句是如何执行的
2021/06/21 MySQL