Centos系统中Pytorch的安装

项目需要Pytorch框架,所以这里记录下他的安装配置过程。一些深度学习框架的安装一般需要适配CUDA版本,【这里需要说明GPU下有了CUDA这个粗GPU加速框架,可以不用cuDNN】,所以本文从CUDA->cuDNN->Pytorch逐一进行配置说明。
1 CUDA配置
  1. 检查自己的设备是否有Nvidia显卡,这个一般方式很多,可以自行查阅相关信息。
  2. 如果硬件设备符合条件,那么就可以上Nvidia_CUDA的官网点击进入:进行查找自己适配的版本,下载【可能需要你注册信息,一般来讲填一个邮箱,或者使用微信、QQ登陆】
  3. 具体的安装流程,参看官方文档点击进入
  4. 如果安装顺利,你可以直接使用 cat /usr/local/cuda/version.txt 该命令查看CUDA版本
2 cuDNN配置
  1. 这个cuDNN框架是主要是针对神经网络的加速,网上有相关的测试,貌似使用该框架,会比正常使用CUDA要快几倍。所以想要加速运算,其实这个框架很有必要。
  2. 进入官网下载相应的版本点击进入【注意】:这里主要和CUDA版本对应。
  3. 下载好后,直接进入安装手册点击进入
  4. 如果顺利,使用 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2 命令可以得到一下信息.Centos系统中Pytorch的安装_第1张图片
  5. 【这里需要注意,一般来讲Linux 环境下的配置需要进入 root 用户才能操作,不能然貌似执行不成功,这如果有人可以在非Root权限下安装,可以私信我,我也想学习!】
3 Pytorch 安装和配置
  1. 有些时候,我们可能会直接去其官网查找对应得CUDA cuDNN 版本的 Pytorch 进行下载安装,如果你的网络支持,欢迎上官网点击进入进行查找安装。
  2. 如果您的网络也是下了一会儿就断了,您可以尝试在清华镜像资源上下载点击进入
  3. 下载后,可以直接进行 到对应的文件夹下 使用命令 :conda install --offline pytorch-*****.tar.bz2 进行离线安装
  4. 安装好 Pytorch 后,再离线进行下载对应的 torchvision ,还是在清华镜像资源上查找就行
  5. 下载后,可以直接进行 到对应的文件夹下 使用命令 :conda install --offline torchvision -*****.tar.bz2 进行离线安装
  6. 这里需要补充一点,以上的安装方式其实少安装了一个 cudatoolkit 包,这个包,可以按照 conda install pytorch=0.4.1 cudatoolkit=9.2 【这里的 0.4.1为pytorch版本,cudatoolkit为cuda版本,均可以根据您的安装需求变化】不安装这个会出现 【CUDA driver version is insufficient for CUDA runtime version】这个错误,具体解决方式可以看我的博文解决方法。
  7. 使用import torch 不报错就好了
    在这里插入图片描述
注意,如果以上步骤您觉得麻烦可以直接到 Pytorch 官网进行傻瓜式的点击图像化界面得到对应系统的安装语句,具体操作点击进入 下拉直到那个下图位置直接可以得到安装Pytorch的语句。

Centos系统中Pytorch的安装_第2张图片
以上如果有问题,欢迎指正!

你可能感兴趣的:(Python,机器学习,目标检测)