Lung-Yu,Tsai 的部落格

以下安裝實測過Ubuntu 16.04 Server 與 Desktop 皆可以成功運作.

自動化安裝之腳本可至github查看最新版本（連結）

先確定環境的Driver以及相關套件安裝完畢.可以透過以下腳本自動安裝.

basic_env.sh

#!/bin/bash
######################################################
# install driver and extras libs on Ubuntu server    #
#####################################################

# NVDRIVER -- NVIDIA driver version from ppa:graphics-drivers/ppa
NVDRIVER='nvidia-384'

# EXTRAS -- Extra packages (your taste may vary)
EXTRAS="build-essential dkms synaptic emacs ssh gdebi" 

# Do system Updates
apt-get update
apt-get -y dist-upgrade

apt-get install -y $EXTRAS 

# NVIDIA driver
add-apt-repository -y ppa:graphics-drivers/ppa
apt-get update
apt-get install -y $NVDRIVER

$ sudo sh basic_env.sh
$ sudo reboot

安裝完畢並重新啟動後,若可以如以下畫面表示基本驅動安裝完成.

$ nvidia-smi

執行自動化安裝docker的腳本後,可以透過sudo systemctl status docker 查看服務狀況,若顯示為activie(running)即代表安裝完成.

$ sudo sh auto_install_docker.sh

$ sudo systemctl status docker

執行自動化安裝腳本後可以自動安裝nvidia-docker,同時會自動進行測試.

若安裝出現如下圖之錯誤訊息；"Could not load UVM kernel module. Is nvidia-modprobe installed?"

表示nvidia-docker 服務沒有正確啟動,請繼續往下安裝其他相依套件.

若無此錯誤訊息則表示應會正常出現顯卡資訊.

sudo sh auto_install_nvidia_docker.sh

透過以下指令可以查看目前nvidia-docker 服務的狀態.其錯誤訊息應會顯示"Failed to start NVIDIA Docker plugin." .

$ sudo systemctl status nvidia-docker

安裝相依套件後,可以透過下述指令啟動,若仍無法正常運行,請reboot(重新啟動).

$ sudo apt-get install nvidia-modprobe

$ sudo systemctl start nvidia-docker
$ sudo systemctl enable nvidia-docker

正常運作後應會如下圖所示：

$ nvidia-docker run --rm nvidia/cuda nvidia-smi