1. 程式人生 > >cuda9.0 + cudnn 7快速安裝

cuda9.0 + cudnn 7快速安裝

前言(廢話)

        又到了tensorflow升級的時候,cuda跟cudnn也都需要升級。但是呢,事物一定是向前發展的,版本升級,功能當然是越來越好啦。cuda + cudnn 的安裝當然也是越來越方便啦。

正文

1. cuda9的安裝

       去官網下好cuda的repo包

sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb

        獲取key

sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub

         更新apt安裝源

sudo apt-get update

        安裝cuda

sudo apt-get install cuda-9-0 

        PS:安裝cuda10.0把上面的cuda-9-0替換為cuda-10-0即可, sudo apt-get install cuda則是安裝最新的cuda版本喲。

2. cudnn的安裝

        下載cudnn的安裝包。

sudo dpkg -i libcudnn7_7.0.5.15-1+cuda9.0_amd64.deb

3.其他問題

(1)libcu**.o is not a symbol link

         出現這個的原因是因為cuda/cudnn中很多.o檔案不是link檔案,而是**.0.version的複製重新命名。

        解決方法,重建link檔案:

        比如 libcupti.so is not a symbol link file.

        我們重建link檔案:

sudo rm libcupti.so libcupti.so.8.0

sudo ln -s libcupti.so.8.0.61 libcupti.so.8.0

sudo ln -s libcupti.so.8.0 libcupti.so

(2)cudnn.o檔案找不到

        這個原因是在7.0版本以上,cudnn不再像以前一樣放在cuda資料夾下,而是單獨的目錄。可以

sudo cp /usr/lib/x86_64-linux-gnu/libcudnn.s* /usr/local/cuda/lib64/

        PS:前提是/usr/local/cuda資料夾是link的/usr/local/cuda-9.0資料夾哈。

(3)cuda9.0切換回cuda8.0版本

        準備工作(只需做一次):

cd /etc/ld.so.conf.d

vi /cuda-9-0.conf

將cuda-9-0替換為cuda,同目錄下(/etc/ld.so.conf.d)有其他cuda*.conf,也要注意,裡面最好link /usr/local/cuda,方便後面的切換。

        開始切換:

cd /usr/local

sudo rm ./cuda

sudo ln -s ./cuda-8-0 cuda

sudo ldconfig -v | grep cu

      搞定,切回9.0同理。