----系統版本：CentOS Linux release 7.7.1908 

----顯示卡：v100

1、新增ELRepo源

$ rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org

$rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-4.el7.elrepo.noarch.rpm

不同作業系統新增源請看此處

 

2、安裝顯示卡檢測程式

$ yum install nvidia-detect

3、檢測顯示卡

$ nvidia-detect -v

Probing for supported NVIDIA devices...
[10de:1cb1] NVIDIA Corporation xxxxxxxxx
This device requires the current 430.40  NVIDIA driver kmod-nvidia
WARNING: Xorg log file /var/log/Xorg.0.log does not exist
WARNING: Unable to determine Xorg ABI compatibility
WARNING: The driver for this device does not support the current Xorg version

4、根據檢測的版本安裝顯示卡驅動程式

$ yum install nvidia-x11-drv-430.40

版本號要一致

6、重啟，執行命令，檢視狀態

$ nvidia-smi

 

安裝完之後 把 tensorflow==1.12這個卸了，然後重新安裝了tensorflow-gpu==1.12，就走GPU進行計算了