ubuntu16安装tensorflow-gpu,官网下的ubuntu16,已安装多台电脑


ubuntu16安装tensorflow-gpu,官网下的ubuntu16,已安装多台电脑_第1张图片
rick

之前安装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

如果没报错的话(一些警告不用管),就大功告成,接下来随便嗨!!


你可能感兴趣的:(ubuntu16安装tensorflow-gpu,官网下的ubuntu16,已安装多台电脑)