机器学习环境配置系列一之CUDA

本文配置的环境为redhat6.9+cuda10.0+cudnn7.3.1+anaonda6.7+theano1.0.0+keras2.2.0+jupyter远程,其中cuda的版本为10.0。

第一步:安装cuda前的工作:

1、检验是否安装了gpu

$ lspci | grep -i nvidia

2、检验redhat版本

$ uname -m && cat /etc/*release

3、检验完毕后就可以根据版本去Nvidia官网下载cuda了。本次安装的操作系统为redhat6.9,显卡为Quadro M6000

  1. 从Nvidia官网下载redhat6版本的cuda(本次选择下载的是run文件)
  2. 用创建的用户(例如MLStudy用户)登陆red hat;
  3. 安装前先要停止Nouveau的运行
    • 创建文件为 /etc/modprobe.d/blacklist-nouveau.conf 内容:如下
blacklist nouveau
options nouveau modeset=0

Regenerate the kernel initramfs:

$ sudo dracut --force
  • 重启系统(reboot),进入Reboot into text mode (runlevel 3). 也可以直接修改 /etc/inittab文件,然后进行重启。
vi /etc/inittab
文件内容

0 - halt (Do NOT set initdefault to this)
# 1 - Single user mode
# 2 - Multiuser, without NFS (The same as 3, if you do not have networking)
# 3 - Full multiuser mode
# 4 - unused
# 5 - X11
# 6 - reboot (Do NOT set initdefault to this)
#
id:5:initdefault:

修改 id:3:initdefault:

    • 运行安装cuda命令
$ sudo sh cuda_<version>_linux.run

  根据提示进行操作就可以完成cuda的安装。

  

原文地址:https://www.cnblogs.com/jaww/p/9845888.html