Ubuntu 深度煉丹環境配置

Ubuntu 深度煉丹環境配置

深度煉丹最麻煩的就是環境配置了，下面過程記錄了本人進行深度煉丹環境的配置。

首先是安裝圖形顯卡驅動，按以下指令進行便可

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install nvidia-396

這樣即可以安裝完圖形顯卡驅動。

緊接着就是安裝CUDA，這是一個比較麻煩的工做。首先是從官網上下載最新的CUDA版本，下面的命令也能夠，可是安裝獲得的版本一般不是最新的。

sudo apt install nvidia-cuda-toolkit

下載runfile便可，安裝過程較爲簡單。爲了驗證安裝是否成功，能夠按以下步驟進行測試：

查看cuda 的版本號，輸入以下指令便可：

nvcc -V

編譯CUDA samples。找到你安裝CUDA的過程當中，選擇的samples保存位置，進入後運行make命令，而後能夠在Samples裏面找到bin/x86_64/linux/release/目錄，並切換到該目錄  運行deviceQuery程序，sudo ./deviceQuery pass即經過。

最後一步最煩人，目前親自試過的比較正確的方式是以下

tar -zxvf cudnn-9.0-linux-x64-v7.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ -d
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*