TensorFlow-GPU环境配置之二——CUDA环境配置

1.安装最新显卡驱动

到系统设置->软件和更新->附加驱动中选中最新的显卡驱动，并应用

TensorFlow-GPU环境配置之二——CUDA环境配置

2.下载CUDA8.0

https://developer.nvidia.com/cuda-downloads（下载地址）

TensorFlow-GPU环境配置之二——CUDA环境配置

3.安装cuDNN6.0

下载地址： https://developer.nvidia.com/cudnn（需要登录）

TensorFlow-GPU环境配置之二——CUDA环境配置

4.安装CUDA8.0

进入下载目录，右键菜单中选择“在终端打开”

TensorFlow-GPU环境配置之二——CUDA环境配置

通过命令安装CUDA8.0

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

TensorFlow-GPU环境配置之二——CUDA环境配置

5.安装cuDNN

gcc降版本：使用g++ --version命令查看版本是6.3.0，然而cuda8.0不支持5.0以上的编译器，因此需要降级，把编译器版本降到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-6 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++-6 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++

TensorFlow-GPU环境配置之二——CUDA环境配置

在下载目录中打开终端，安装cuDNN

tar xvzf cudnn-8.0-linux-x64-v6.0-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*

TensorFlow-GPU环境配置之二——CUDA环境配置

6.配置环境变量

打开bash_profile

sudo gedit ~/.bash_profile

在打开的文本末尾加入

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

保存退出后通过如下命令使环境变量生效

source ~/.bash_profile

原文链接: https://www.cnblogs.com/conorpai/p/6946051.html

欢迎关注

微信关注下方公众号，第一时间获取干货硬货；公众号内回复【pdf】免费获取数百本计算机经典书籍

原创文章受到原创版权保护。转载请注明出处：https://www.ccppcoding.com/archives/254839

非原创文章文中已经注明原地址，如有侵权，联系删除

关注公众号【高性能架构探索】，第一时间获取最新文章

转载文章受原作者版权保护。转载请注明原作者出处！

TensorFlow-GPU环境配置之二——CUDA环境配置

相关推荐