之前安装tensorflow-gpu遇到很多坑,贡献一个成功的过程。
cuda和cudnn可以官网下,我贴一个我下的链接: https://pan.baidu.com/s/1eSelr0Y 密码: 6q2m
1, 降级gcc 降到4.9
sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++
2, 安装nvdia驱动
设置->软件和更新->驱动->nvidia
3, 安装cuda
ctrl+c 跳过介绍
install NVIDIA Driver 选 n(这个是让你安装cuda自带的驱动,第二步已经安装过了,所以选否,不然可能会出问题)
配置环境变量(我喜欢用~目录下的.bashrc, 记得source .bashrc, 不然会提示找不到cuda)
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/ extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda
4, 安装cudnn
tar -zxvf cudnn*.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
5, pip install tensorflow-gpu
6, 进入python, 输入import tensorflow as tf
如果没报错的话(一些警告不用管),就大功告成,接下来随便嗨!!