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

原文地址:https://www.cnblogs.com/lpfworld/p/9329562.html