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

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容