zoukankan      html  css  js  c++  java
  • pytorch入门之安装和配置

    pytorch是一种python接口的深度学习框架,其他的框架还有caffe,tensorflow等等。

    1,pytorch目前支持linux和OSX两种系统。支持的Python版本有2.7,3.5,3.6。

    2,包管理工具PackageManager我们选择conda,在Anaconda这个功能强大的包中包含了conda,也包含python以及很多python的扩展工具包。在需要的时候,选择下载安装就可以了。

    3,学习深度学习最好有支持cuda的NVIDIA显卡,用来进行快速的计算。

          在这里我们需要做的是:

         一,安装显卡驱动,在Ubuntu上,在SystemSettings--》Software&Updates--》AdditionDrivers中联网状态下,会自动搜索可以使用的显卡,点击ApplyChanges,重启电脑,显卡驱动生效。

         二,安装CUDATookit,pytorch最多支持到CUDA8.0,要在官网找到相应的下载包括了cuda8.0和对应的cudnn的安装。

              安装完以后要进行相关的配置,在home目录中的.bashrc文件中,在末尾添加一下代码:

              export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
              export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
              export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

    4, 到现在才是真正的安装pytorch

         安装pytorch需要安装两个东西,pytorch和torchvision。用相关命令安装即可

    5,验证安装是否成功

         sudo lshw -c video 查看显卡信息

        nvcc -V  显示CUDA信息

        python进入python环境,import  torch   import  torchvision 不报错就是成功了

  • 相关阅读:
    Chrome(google 浏览器) 几个有用的插件
    unity创建Android原生插件
    C#中运算符
    Unity 动画播完时关闭动画
    Could not obtain connection to query metadata : An attempt by a client to checkout a Connection has timed out.]
    nginx将ip+端口号映射为域名
    GlassFish To prevent a memory leak, the JDBC Driver has been forcibly unregistered
    视频
    iOS App上架流程(2016详细版
    证书说明(二)
  • 原文地址:https://www.cnblogs.com/lpfworld/p/9329562.html
Copyright © 2011-2022 走看看