centos 7安裝nvidia顯卡驅動和CUDA

# 查看本機顯卡
lspci | grep -i nvidia

# 添加ELRpo源
# 1. 導入公鑰
rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
# 2. 添加源
rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-2.el7.elrepo.noarch.rpm

# 安裝顯卡檢查程序
yum -y install nvidia-detect
# 查找合適驅動, 一般是kmod-nvidia
nvidia-detect

# 安裝kernel-devel依賴, 沒有gcc的還要安裝gcc
yum install -y kernel-devel-$(uname -r) kernel-headers-$(uname -r)

# 安裝顯卡驅動 install $(nvidia-detect)
yum -y install kmod-nvidia

# 重啓
reboot

# 驅動安裝完成
nvidia-smi

# 切換到tmp
cd /tmp/
# 下載cuda, 版本下載鏈接可去nvidia官網查
wget http://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm

# 安裝cuda
rpm -i cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm
yum clean all
yum -y install cuda

# 添加環境變量
export PATH=$PATH:/usr/local/cuda-10.2/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64:$LD_LIBRARY_PATH

# 查看cuda是否安裝成功
nvcc -V
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章