image
目前最新的 Ubuntu Server 16.04.3 LTS 系統(tǒng)為例,介紹包括 CUDA、CudNN 軟件以及 TensorFlow、Keras 等 Python 庫的安裝方法。
下面是需要安裝的軟件列表。
- Cuda
- CudNN
- TensorFlow, Pytorch
- OpenCV,Caffe
- Matlab
前期準(zhǔn)備: sudo apt-get update, 找到清華大學(xué)鏡像站,下好anaconda并安裝,然后將anaconda的python添加到環(huán)境變量
然后下載到NVIDIA下載cuda
image
一個坑: 必須<meta charset="utf-8">關(guān)閉圖形界面,不然會報(bào)錯
sudo service lightdm stop
image
cuda安裝好
image
下面安裝cuDNN, 用來加速深度學(xué)習(xí)框架
到英偉達(dá)官網(wǎng)下載好壓縮包,然后解壓縮。將cuda/lib64的文件復(fù)制到 /usr/local/cuda/lib64中,cuda/include/cudnn.h 復(fù)制到 /usr/local/cuda/include
然后更新libcudnn.so.7 連接, 然后ldconfig --> <meta charset="utf-8">搜索出可共享的動態(tài)鏈接庫(格式如lib.so),進(jìn)而創(chuàng)建出動態(tài)裝入程序(ld.so)所需的連接和緩存文件。
sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
sudo ln -sf libcudnn.so.7 libcudnn.so
sudo ldconfig -v
到NVIDIA給的sample 里面測試一下
cd samples && make
等待一會之后,編譯完成。
./bandwidthtest
image