美文网首页
Pytorch1.0安装

Pytorch1.0安装

作者: Latet | 来源:发表于2019-02-14 18:57 被阅读0次

    前言

    本文记录了在windows下配置pytorch环境的步骤。如需转载,请附出处https://www.jianshu.com/p/07abfe5f82e1

    1、Anaconda安装

    Anaconda安装,去官网下载比较慢,所以选取清华源镜像链接下的安装包,链接:Anaconda安装包下载
    安装好之后打开anaconda navigator,效果如下:

    anaconda.png
    其中最重要的就是查看环境(environments),最先安装好时只会有一个base环境,我安装的是基于python3.6的base环境的anaconda
    conda包管理命令
    ## 创建指定python3.6版本的环境,环境名为py36
    conda create –n py36 python=3.6
    ## 查看安装的环境名
    conda info --envs
    ## 激活环境名为py36的环境
    activate py36
    

    查看安装环境效果:
    带星号的代表当前所处环境
    图中显示一共有三个环境,root代表默认环境,另外两个代表新建环境

    2、pytorch安装(CPU)

    pytorch主页提供了安装的命令操作,根据是否有GPU和cuda版本,安装具体版本的pytorch都做了显示。


    但是,在安装pytorch和一些基本库(numpy、matplotlib等)时,有些时候网速会特别慢,这个时候安装会采用设置清华源镜像路径来安装。
    • 如果是conda安装,则安装前输入以下代码
    ### 设置清华源镜像
     conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
     conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
     conda config --set show_channel_urls yes
    ### 设置pytorch镜像
     conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
     conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
    

    添加镜像后将官方安装的命令-c pytorch去掉会快的飞起

    • 如果是pip安装,则安装前输入以下代码
     pip install -i https://pypi.tuna.tsinghua.edu.cn/simple name
    

    另外,老版本的镜像中不存在最新版的pytorch1.0,所以采取镜像安装pytorch显示来源也是pytorch官网,针对未翻墙的用户还是特别忙。当然,最适用的办法是下载离线包,然后再进行安装。离线包下载链接

    1.png
    当然,也可以根据上面的下载链接里的badges里的链接来设置pytorch镜像链接。下载好的离线安装包直接采用conda install name就好了
    如下图,安装成功!

    3、pytorch安装(GPU)

    • 首先,如果电脑有支持CUDA的N卡的话,就可以安装GPU版的pytorch进行GPU加速。CUDA9.1下载链接,我安装的是CUDA9.1,安装过程中记得关闭360软件,不然会一直提示中毒。安装过程中屏幕会闪烁,是正常现象,安装好之后在CMD里面输入nvcc -V就会显示安装的CUDA版本
    • 在安装好CUDA之后,需要安装支持的cudnn(GPU加速深度学习包),cudnn下载链接,根据对应得CUDA来安装对应得cudnn,我安装的是cuDNN v7.3.0 Library for wins10,然后将解压的三个文件夹复制在CUDA安装位置。
      1. 下载好的安装包


      2. 解压


      3. 将解压的三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.1里面


    • 这样,就安装好了CUDA和cudnn了,然后按照pytorch cpu版安装的步骤来安装GPU版的pytorch
    conda install pytorch cuda91
    pip install torchvision
    

    安装检查:显示cuda可用,pytorch版本1.0


    相关文章

      网友评论

          本文标题:Pytorch1.0安装

          本文链接:https://www.haomeiwen.com/subject/ntxyeqtx.html