1. 程式人生 > >機器學習環境配置系列一之CUDA

機器學習環境配置系列一之CUDA

arch developer 完成 用戶 uname sed a10 是否 安裝

本文配置的環境為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的安裝。

  

機器學習環境配置系列一之CUDA