機器學習環境配置系列一之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的安裝。

  


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM