如何搭建pytorch环境的方法步骤

 更新时间:2020-05-13 06:45:35   作者:佚名   我要评论(0)

1.conda创建虚拟环境pytorch_gpu


conda create -n pytorch_gpu python=3.6

创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无需很长,

1.conda创建虚拟环境pytorch_gpu

conda create -n pytorch_gpu python=3.6

创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无需很长,成功之后界面如下所示:

2.切换到pytorch环境

使用如下命令,切换到我们刚刚创建好的pytorch虚拟环境,这样我们避免与其它python环境之间的干扰。

conda activeta pytorch_gpu

切换成功之后就会看到在路径前边显示我们已经进入该虚拟环境。

3.安装几个常用库(也可暂时不安)

conda install pandas jupyter notebook

4.安装pytorch

4.1进入官网查看要下载的版本

查看对应的版本,这里是官方链接:

4.2 根据系 统信息及cuda版本选择对应toolkit

这里最主要的是那个CUDA的版本,此处我选择的是10.1,是因为我的电脑的 cuda版本信息就是这样的。具体的查看方法可在4.5节查阅。

4.3复制上图中最后一行代码到pytorch环境终端

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

这里是下载过程截图:


但是在下载过程中torchvision以及pytorch没有下载成功,


因为这是去其官网下载,下载速度很慢,在上图中我们也可以看出是因为网络错误,网络上也有几种其他的方法,此处我没去验证,我还是让电脑重新下载的,等待时间挺长的,但是因为是在晚上下载的,一早起来就好了

4.4 验证pytorch是否安装成功

此时直接输入 python,即可成功进入:


而后输入如下指令,查看torch是否安装成功

>>> import torch
>>> x=torch.randn(4,4)
>>> print(x)

正常情况下是出现这个界面的:


验证完成之后,可以quit()保存退出。

4.5 如何查看自己电脑cuda版本

4.5.1 windows如何查看

NVDIA控制面板–>帮助–>系统信息


组件–>NVCUDA.DLL 可以查看CUDA版本


这里我的显示是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搭建环境内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!

您可能感兴趣的文章:

  • 使用pytorch搭建AlexNet操作(微调预训练模型及手动搭建)
  • 关于pytorch中全连接神经网络搭建两种模式详解
  • Pytorch 搭建分类回归神经网络并用GPU进行加速的例子
  • 运用PyTorch动手搭建一个共享单车预测器
  • PyTorch搭建一维线性回归模型(二)
  • PyTorch搭建多项式回归模型(三)
  • PyTorch快速搭建神经网络及其保存提取方法详解
  • PyTorch上搭建简单神经网络实现回归和分类的示例

相关文章

  • 如何搭建pytorch环境的方法步骤

    如何搭建pytorch环境的方法步骤

    1.conda创建虚拟环境pytorch_gpu conda create -n pytorch_gpu python=3.6 创建虚拟环境还是相对较快的,它会自动为本环境安装一些基本的库,等待时间无需很长,
    2020-05-13
  • Python打印特殊符号及对应编码解析

    Python打印特殊符号及对应编码解析

    1、调用字符映射表输入特殊符号 在键盘上按win+R,在打开的对话框中输入“charmap”,会出现字符映射表: 2、利用字符编码输入特殊符号 #打印Σ print(chr(931))
    2020-05-13
  • Python3自动生成MySQL数据字典的markdown文本的实现

    Python3自动生成MySQL数据字典的markdown文本的实现

    为啥要写这个脚本 五一前的准备下班的时候,看到同事为了做数据库的某个表的数据字典,在做一个复杂的人工操作,就是一个字段一个字段的纯手撸,那速度可想而知是
    2020-05-13
  • 解决pycharm安装第三方库失败的问题

    解决pycharm安装第三方库失败的问题

    前言: 在使用pycharm学习python的时候,经常需要第三方库,没有第三方库程序就会报错,pycharm也会提醒你要安装所需要的库,安装第三方库的时候往往就出现了问题,
    2020-05-13
  • Python使用sqlite3模块内置数据库

    Python使用sqlite3模块内置数据库

    1、python内置的sqlite3模块,创建数据库中的表,并向表中插入数据,从表中取出所有行,以及输出行的数量。 #!/usr/bin/env python3 #创建SQLite3内存数据库,并创
    2020-05-13
  • 基于python实现检索标记敏感词并输出

    基于python实现检索标记敏感词并输出

     一些带有过度宣传的词,在淘宝、京东对商品的宣传有一定的限制,这些最佳,最大、盗版、水货等词语都不能用于产品的宣传,可以使用程序检测敏感词,以下既是具体代
    2020-05-13
  • Python接口测试数据库封装实现原理

    Python接口测试数据库封装实现原理

    引言   做接口测试的时候,避免不了操作数据库。因为数据校验需要,测试数据初始化需要、一些参数化场景需要等。   数据库操作框架设计   这里主要操作mysql数
    2020-05-13
  • Python reduce函数作用及实例解析

    Python reduce函数作用及实例解析

    语法 在python3中,内置函数中已经没有reduce了。要使用reduce,需要从functools模块里引入 可以看到,reduce有三个参数,第一个是函数function,第二个是序列sequ
    2020-05-13
  • python多进程下的生产者和消费者模型

    python多进程下的生产者和消费者模型

    一、生产者消费者模型介绍 1.1 为什么需要使用生产者消费者模型 生产者是指生产数据的任务,消费者是指消费数据的任务。当生产者的生产能力远大于消费者的消费能力,
    2020-05-13
  • win10从零安装配置pytorch全过程图文详解

    win10从零安装配置pytorch全过程图文详解

    1.安装anaconda (anaconda内置python在内的许多package,所以不用另外下载python) 可以点击下面的清华开源软件镜像站,在官网下载anaconda不如在这下的快 htt
    2020-05-13

最新评论