机器学习环境配置系列一之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
- 从Nvidia官网下载redhat6版本的cuda(本次选择下载的是run文件)
- 用创建的用户(例如MLStudy用户)登陆red hat;
- 安装前先要停止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的安装。