CentOS 8 下安裝NVIDIA RTX 3070 驅動程式及CUDA

woff

下載檔案安裝
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