cuda9.0 + cudnn 7快速安裝
阿新 • • 發佈:2018-12-17
前言(廢話)
又到了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同理。