一、安装anaconda

 anaconda安装简单,只要确定自己的系统即可,具体安装请参考这里

二、确定自己的系统版本

 我的是centos

cat /etc/redhat-release

 查看linux系统方法:

# 方法一
cat /proc/version# 方法二
lsb_release -a# 方法三
cat /etc/issue# 方法四
cat /etc/redhat-release

三、确定自己的CUDA版本

 什么是cuda:

  CUDA(ComputeUnified Device Architecture),是显卡厂商NVIDIA推出的运算平台。 CUDA是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题

  简单说强大的芯片如果只是作为显卡就太浪费了,因此N卡厂商推出CUDA,让显卡可以用于图像计算以外的目的

 什么是cudnn:(注意:CUDA和CuDNN一般都是结伴而行,如果需要CUDA,必然会需要CuDNN)

 下载地址:注意这里需要注册才能下载,我用qq邮箱,密码首字母大写

  CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手

  但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多

  cudnn下载解压后cuda目录下有下面内容:

  修改权限:

sudo cp include/cudnn.h /usr/local/cuda/include
sudo cp lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

 Centos7安装NVIDIA显卡驱动和CUDA10.1(以10.1为例):

  先查看自己的linux上显卡型号:

lspci | grep -i nvidia

  可见我的显卡杠杠滴

  显卡驱动下载:(下载与自己显卡型号一致的驱动)

  注意:在安装显卡驱动的时候如果报错

ERROR: An NVIDIA kernel module 'nvidia-uvm' appears to already be loaded in your kernel.  This may be because it is in use (for example, by the X server), but may also happen if your kernel was configured

  我们需要查看是否有程序占用(如果存在占用,请停掉该程序)

lsof | grep nvidia.uvm

  当然显卡驱动也可以这样安装:(推荐)

sudo yum install nvidia-detect  # 安装检查显卡的程序
nvidia-detect -v  # 查看需要的显卡驱动型号
yum -y install kmod-nvidia  # 安装驱动

  卸载驱动:

sudo yum remove kmod-nvidia

  

  如果通过官网下载驱动安装时找不到kernel,通过下面的方式安装驱动(确保kernel已经安装)

sudo ./NVIDIA-Linux-x86_64-430.40.run --kernel-source-path=/usr/src/kernels/3.10.0-514.el7.x86_64

  

  cuda下载:

  

  检查linux是否安装了GPU

lspci | grep -i nvidia

  安装 cuda 的时候,会询问是否安装显卡驱动,说明 cuda 安装程序里包含了的显卡驱动;建议先不要安装 cuda 里的显卡驱动,待安装完 cuda 后,执行例子程序,如果报错再检查显卡驱动是否正确,避免覆盖原来的显卡驱动

  安装gcc、g++编译器:先查看是否已经安装(gcc -v)

yum install gcc
yum install g++

  安装kernel-devel和kernel-headers:

yum install kernel-devel
yum install kernel-headers

  注意这里的kernel要和自己的linux系统版本一致:可以使用uname -r查看自己的系统版本

uname -r
#3.10.0-514.el7.x86_64

  有时候直接yum安装的版本比较新,导致显卡驱动安装失败

sudo yum remove kernel-devel # 卸载新的
sudo rpm -i kernel-devel-3.10.0-514.el7.x86_64.rpm #下载安装适合的版本

  版本下载

  编辑grub文件

vim /etc/default/grub

  在GRUB_CMDLINE_LINUX中的末尾(注意要包含在引号内)添加/rd.driver.blacklist=nouveau nouveau.modeset=0

  随后生成配置:

grub2-mkconfig -o /boot/grub2/grub.cfg

  

  禁用nouveau:将blacklist nvidiafb注释掉,并添加如下内容,禁用的原因点击这里

   blacklist nouveau
   options nouveau modeset=0

  更新配置并重启:

sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)
reboot

  

  重启后确认禁用了nouveau:

lsmod | grep nouveau
# 没有内容输出表示禁用成功

  

  赋予权限并安装驱动和CUDA-10.0:

./NNVIDIA-Linux-x86_64-430.50.run
./cuda_10.1.243_418.87.00_linux.run

  查看安装结果

nvidia-smi

  显卡驱动和cuda安装成功后,就可以用conda install 来安装python包了(注意:虚拟机无法调用GPU,即使所有的环境配置好,代码依然会报:all cuda 忙或者不可用的错误)

  推荐搭建使用docker来安装深度学习环境,方便部署

  总结:大家可以参考这个来安装深度学习框架

转载于:https://www.cnblogs.com/always-fight/p/11512608.html

centos下通过conda安装pytorch相关推荐

  1. linux安装pytorch教程,centos下通过conda安装pytorch

    一.安装anaconda anaconda安装简单,只要确定自己的系统即可, 二.确定自己的系统版本 我的是centos cat /etc/redhat-release 查看linux系统方法: # ...

  2. 使用conda卸载pytorch_centos下通过conda安装pytorch

    一.安装anaconda anaconda安装简单,只要确定自己的系统即可, 二.确定自己的系统版本 我的是centos cat /etc/redhat-release 查看linux系统方法: # ...

  3. linux pytorch环境配置,linux下使用conda安装pytorch,并配置pytorch

    8种机械键盘轴体对比 本人程序员,要买一个写代码的键盘,请问红轴和茶轴怎么选? 首先安装anaconda 去官网下载.sh包,安装之后会把jupyter一起安装好. 配置jupyter 首先先生成一个 ...

  4. conda安装pytorch下载太慢

    导读 在使用conda安装pytorch的时候下载特别慢,有时候下载到一半了还会断掉,然后还需要重新下载才行,很难受. 比如我们要安装pytorch1.3.1的版本,安装命令如下 conda inst ...

  5. 【conda安装pytorch总是下载cpu版本的问题】

    conda安装pytorch总是下载cpu版本的问题 首先,表述问题:我在使用pytorchu官方网站安装torch1.9.0的GPU包, 链接: link 确保自己的安装的代码为 // cuda 1 ...

  6. 如何解决CentOS下使用yum安装python-pip失败

    用Ubuntu的时候感觉很简单的事到ContOS上却变得很头痛,CentOS下使用yum安装python-pip失败的完美解决方法. 在执行以下命令安装Python-pip居然什么也没执行 yum i ...

  7. CentOS下Redis的安装

    CentOS下Redis的安装 前言 安装Redis需要知道自己需要哪个版本,有针对性的安装,比如如果需要redis GEO这个地理集合的特性,那么redis版本就不能低于3.2版本,由于这个特性是3 ...

  8. Centos下gcc的安装、gcc的更新、gcc安装过程中报错:make[1]: *** [stage1-bubble] 错误 2

    Centos下gcc的安装.gcc的更新.gcc安装过程中报错:make[1]: *** [stage1-bubble] 错误 2 文章目录: 1 下载gcc的源码 1.1 源码下载地址: 1.2 下 ...

  9. CentOS下软件的安装

    Linux操作系统上软件的安装从整体上来说有两种方法: 一种是二进制安装,另一种是源码安装 下面我们就先说一下二进制安装 二进制安装我们一般采用两种工具rpm工具和yum工具 1.rpm工具 rpm ...

最新文章

  1. uboot2014.10移植(一)
  2. C#多线程学习之(五)使用定时器进行多线程的自动管理
  3. 二叉树的的前序遍历和后序遍历(题型变种更新中)
  4. SAP CRM WebClient UI F4 value help is centrally implemented in CRM_THTMLB_UTIL/F4Frameset.html
  5. 播放2.0:Akka,Rest,Json和依赖项
  6. php 面试靠快速排序,搞定PHP面试 - 常见排序算法及PHP实现
  7. php mysql ajax登录界面_PHP+jQuery+Ajax实现用户登录与退出
  8. 在nodeJs的Express框架下用TypeScript编写router路由出现import关键字错误的解决方案
  9. PyTorch 和 TensorFlow 引领未来
  10. Java常用中间件---SpringBoot整合ActivityMQ
  11. 58同城峰会落幕 智能化平台和下沉市场能让继续神奇吗?
  12. stm32零基础应该怎么入门?
  13. php中的oop思想,php oop的特性是什么?php oop思想的三大特点介绍
  14. 气球java游戏_气球游戏-2019腾讯笔试
  15. linux 桌面 v2ex,程序员:他人笑我桌面太凌乱,我笑他人看不穿
  16. 10004 - Bicoloring
  17. linux服务器上装r,linux服务器安装R语言及Rstudio server
  18. 第五周学习总结-HTML5
  19. Excel建立股票资产定价模型,寻找最佳投资组合
  20. 如何优雅地蹭邻居家的wifi

热门文章

  1. js 能实现监听F5页面刷新子iframe 而父页面不刷新
  2. 计算机:游戏14,英语专四作文满分范文(十四):电脑游戏
  3. android中进行https连接的方式的详解
  4. sqlserver2005 openRowSet 和row_Number
  5. celery cluser redis_celery结合redis 使用
  6. 自走棋投降代码_自走棋太火惨遭针对!代码哥之后又是锁钱挂?玩家:还有拒绝员!...
  7. web开发时浏览器兼容性问题汇总
  8. 使用Python在ArcGIS中编程杂谈
  9. 程序转制作人——游戏设计课程导论
  10. java中for 的几种常见用法