深度学习 环境相关(持续更新)

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

你可能感兴趣的:(深度学习 环境相关(持续更新))