3. cudnn升級
最近tensorflow-gpu版本升級了, 原來要求使用cudnn5, 現在需要cudnn6, 于是乎到nvidia官網重新安裝, 下載cudnn-8.0-linux-x64-v6.0.tgz這個包, 解壓 tar -xvf cudnn-8.0-linux-x64-v6.0.tgz
, 分別拷貝文件到對應目錄
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, 是個virtualEnv的'site' has no attribute "getsitepackages"的bug, 暫時用最粗暴的方式解決, 在外面安裝一個tensorflow和tensorboard