ubuntu 18.04 64bit如何安装GPU版本tensorflow

转载

mb5fd868b989ae9 2019-01-15 00:26:00

文章标签 tensorflow python linux g++ ubuntu 文章分类 代码人生

注:笔者的ubuntu18.04 64bit已经安装好了显卡驱动，因此没有此步操作

1.获取cuda(https://developer.nvidia.com/cuda-downloads,选择ubuntu相关的runfile,cuda_9.0.176_384.81_linux.run)

2.安装cuda

sudo apt-get install gcc-6 g++-6 (cuda 9仅支持gcc-6,而ubuntu18.04 64bit默认版本为gcc-7.3.0)

cd /usr/bin
sudo mv gcc gcc-7.3.0.back
sudo ln -s gcc-6 gcc
sudo mv g++ g++-7.3.0.back
sudo ln -s g++-6 g++

chmod u+x cuda_9.0.176_384.81_linux.run

sudo ./cuda_9.0.176_384.81_linux.run (配置时不要安装驱动,Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81? (y)es/(n)o/(q)uit: n)

将/usr/local/cuda-9.0/bin加入到环境变量PATH中

将/usr/local/cuda-9.0/lib64添加到/etc/ld.so.conf中,并且运行ldconfig进行配置

3.获取cudnn(https://developer.nvidia.com/rdp/cudnn-download,需要注册,选择archive版本cudnn-9.0-linux-x64-v7.4.1.5.tgz,为什么没有选择cudnn for ubuntu18.04这样的版本?因为这个版本太高,目前tensorflow-gpu只支持到cuda9.0,而cudnn必须与cuda版本匹配)

4.安装cudnn

tar xvf cudnn-9.0-linux-x64-v7.4.1.5.tgz

cd cuda (没错，目录名为cuda)

sudo cp include/* /usr/local/cuda/include/

sudo cp lib64/* /usr/local/cuda/lib64/

5.安装python相关软件

sudo apt-get install python-pip python-dev python3-pip python3-dev

6.安装tensorflow(tensorflow的版本需要与cuda以及cudnn匹配)

pip3 install tensorflow-gpu==1.12.0

7.测试

#python3

>>> import tensorflow as tf (没有错误信息表明基本安装ok)