TShopping

標題: CentOS 8 下安裝NVIDIA RTX 3070 驅動程式及CUDA [打印本頁]
作者: woff    時間: 2024-3-20 18:35
標題: CentOS 8 下安裝NVIDIA RTX 3070 驅動程式及CUDA
下載檔案安裝
1. 先到NVIDIA下載驅動程式NVIDIA-Linux-x86_64-470.239.06.run


2. 安裝kernel-headers
  1. dnf install kernel-devel-$(uname -r) kernel-headers-$(uname -r)
複製代碼
3. 安裝GCC、kernal組件、dkms等
  1. yum install gcc
  2. yum install gcc-c++
  3. yum -y install kernel-devel
  4. yum -y install kernel-headers
  5. yum -y install epel-release
  6. yum -y install dkms
複製代碼


網路安裝 需安裝 2-3步驟
  1. dnf config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/cuda-rhel8.repo
複製代碼
  1. lspci | grep VGA
複製代碼
  1. dnf module list nvidia-driver
複製代碼

  1. dnf module -y install nvidia-driver:latest-dkms
複製代碼
  1. nvidia-modprobe && nvidia-modprobe -u
複製代碼
  1. reboot
複製代碼


安裝完成後，檢查會顯示
  1. nvidia-smi
  2. dmesg | grep NVRM
複製代碼


4. 下載CUDA並安裝CUDA
  1. wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-rhel8-11-7-local-11.7.0_515.43.04-1.x86_64.rpm
  2. rpm -i cuda-repo-rhel8-11-7-local-11.7.0_515.43.04-1.x86_64.rpm
  3. yum clean all
  4. yum -y install nvidia-driver-latest-dkms cuda
  5. yum -y install cuda-drivers
複製代碼

5. 網路安裝
  1. dnf -y install cuda-11-1
複製代碼


6. CUDA導出為環境變數
開啟vim進行編輯。

  1. vi ~/.bashrc
複製代碼

增加以下內容：

  1. export PATH=/usr/local/cuda/bin:$PATH
  2. export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
複製代碼


修改完成之後退出編輯模式，在尾行模式下儲存退出。

  1. ：wq
複製代碼


在目前shell中執行，使環境變數生效。

  1. source ~/.bashrc
複製代碼


6. 查看是否安裝成功
  1. nvcc -V
  2. nvidia-smi
複製代碼


參考文章https://www.server-world.info/en/note?os=CentOS_Stream_8&p=cuda&f=2#google_vignette
https://developer.nvidia.com/cud ... rget_type=rpm_local
https://blog.csdn.net/weixin_41650348/article/details/115110021









歡迎光臨 TShopping (http://www.tshopping.com.tw/) Powered by Discuz! X3.2