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 不报错就是成功了

  • 相关阅读:
    JavaScript中的方法和属性
    vue的计算属性与方法的不同
    vue 中使用 watch 出现了如下的报错
    vue中methods函数调用methods函数写法
    ES2015箭头函数与普通函数对比理解
    leetcode Single Number python
    leetcode first bad version python
    leetcode Search for a Range python
    leetcode Longest Valid Parentheses python
    python dict traversal
  • 原文地址:https://www.cnblogs.com/lpfworld/p/9329562.html
Copyright © 2011-2022 走看看