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

  • 相关阅读:
    计算机程序的思维逻辑 (10)
    计算机程序的思维逻辑 (8)
    计算机程序的思维逻辑 (9)
    计算机程序的思维逻辑 (7)
    计算机程序的思维逻辑 (6)
    计算机程序的思维逻辑 (5)
    计算机程序的思维逻辑 (4)
    计算机程序的思维逻辑 (3)
    数字电路设计之STA
    数字电路-亚稳态机制
  • 原文地址:https://www.cnblogs.com/lpfworld/p/9329562.html
Copyright © 2020-2023  润新知