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

Posted 2021-01-14 jaww

本文配置的环境为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

• • 重启系统（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的安装。