我这里配置的是tensorflow1.6.0 + cuda9.0 + cudann 7.1.1
1.更改驱动
在“软件和更新”里找到 附加驱动把驱动换成NIVIDA BINARY 我的是384.111
2.安装cuda
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal
这里是cuda9.0的下载地址,英伟达有对应的安装指令 输入即可
3.安装cudann
登录后在这里下载 对应打这个
cuDNN v7.1.1 Library for Linux
用tar指令解压
然后
sudo cp cuda/ include/cudnn.h /usr/local/cuda/include
sudo cp cuda/ lib64/libcudnn* /usr/local/cuda/lib64
最后sudo gedit ~/.bashrc
加上
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64"
export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin:$PATH
4.装好tensorflowGPU环境 就好了
视频教程:
https://www.youtube.com/watch?v=fY0n0WD3UXs