3. cudnn升級(jí)
最近tensorflow-gpu版本升級(jí)了, 原來(lái)要求使用cudnn5, 現(xiàn)在需要cudnn6, 于是乎到nvidia官網(wǎng)重新安裝, 下載cudnn-8.0-linux-x64-v6.0.tgz這個(gè)包, 解壓 tar -xvf cudnn-8.0-linux-x64-v6.0.tgz
, 分別拷貝文件到對(duì)應(yīng)目錄
sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn.so.6.0.21 /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*
cd /usr/local/cuda-8.0/lib64
sudo ln -sf libcudnn.so.6.0.21 libcudnn.so.6
sudo ln -sf libcudnn.so.6 libcudnn.so
sudo ldconfig
4. tensorflow
tensorflow輔助
$ pip install tensorboard
注意在virtualEnv下安裝有的bug, 是個(gè)virtualEnv的'site' has no attribute "getsitepackages"的bug, 暫時(shí)用最粗暴的方式解決, 在外面安裝一個(gè)tensorflow和tensorboard