unbuntu18.0.4 安裝nvidia驅動 cuda cudnn

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系統環境變量配置

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章