1.打開終端窗口,使用以下命令檢查nvidia卡正在使用哪個驅動程序:
sudo lshw -c display
還可以使用video而不是display作爲類名:
sudo lshw -c video
默認情況下,開源nouveau驅動程序用於Nvidia卡
2.禁用 nouveau 顯卡驅動
/etc/modprobe.d/blacklist.conf裏添加如下內容
blacklist nouveau
options nouveau modeset=02.
重新生成 kernel initramfs:
sudo update-initramfs -u
3.安裝nvidia驅動
查看設備可用驅動
sudo ubuntu-drivers devices
安裝所有推薦驅動
sudo ubuntu-drivers autoinstall
安裝指定版本的nvidia驅動
sudo apt install nvidia-driver-version-number
例如:
sudo apt install nvidia-driver-440
安裝驅動程序後,我們需要重新啓動計算機以啓用nvidia-prime,可在Intel和Nvidia顯卡之間切換:
sudo shutdown -r now
查看當前使用的nvidia驅動程序:
sudo lshw -c display
檢查現在正在使用的卡
prime-select query
如果要使用Intel圖形卡,請運行以下命令:
sudo prime-select intel
要切換回Nvidia卡,請運行
sudo prime-select nvidia
安裝完驅動後,需要重新啓動Ubuntu 18.04系統才能生效。
4. 安裝cuda
cuda 歷史版本列表(先確定和tensorflow 版本 以及nvidia驅動版本一致)
https://developer.nvidia.com/cuda-toolkit-archive
安裝指南
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
5. 安裝cudnn
cudnn 歷史版本列表(先確定和tensorflow 版本 以及nvidia驅動版本一致)
https://developer.nvidia.com/rdp/cudnn-archive
如果找不到命令請配置環境變量到 /etc/profile( 重啓系統纔可永久生效 ) 或者用戶環境變量(linux系統環境變量配置)