Ubuntu16.04 CUDA CUDNN Anaconda TensorFlow安装

最新推荐文章于 2021-08-13 14:07:45 发布

哀酱

最新推荐文章于 2021-08-13 14:07:45 发布

阅读量738

点赞数

分类专栏：深度学习文章标签： ubuntu cuda tensorflow

本文链接：https://blog.csdn.net/u010821666/article/details/78982057

版权

深度学习专栏收录该内容

13 篇文章 4 订阅

订阅专栏

我的经验是～～所有的东西，最好都看原版的官方的文档，这样最准确高效。网上的博客很难找到合适的。

安装cuda 8.0

按照官方文档！！！http://docs.nvidia.com/cuda/cuda-installation-guide-linux/#axzz4HIBXnwyt

检查

显卡

lspci | grep -i nvidia

这里写图片描述

系统

uname -m && cat /etc/*release

这里写图片描述

gcc

gcc --version

这里写图片描述
保证5.0以上

使用deb方式安装

简单高效，也有人说会出很多问题，但我还没有遇到过。
重装完系统，还没有驱动（nvidia-smi命令找不到），直接使用deb方式安装
cuda toolkit 8.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

三条命令结束战斗。

现在查看驱动是有信息的

nvidia-smi

详细信息使用

nvidia-smi -a

这里写图片描述

查看cuda是否安装成功

cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery

显示一些GPU相关的信息就证明安装成功了
这里写图片描述

安装anaconda

强烈推荐清华的镜像站！！从官网上下特别特别慢
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
下载Anaconda2-4.4.0-Linux-x86_64.sh

bash Anaconda2-4.4.0-Linux-x86_64.sh

安装cuDNN

从官方网站上下载cuDNNnvidia cuDNN 对应cuda8.0

cd /usr/local
sudo tar -xzvf ~/cudnn-8.0-linux-x64-v5.0-ga-tgz

vim ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-8.0
export PATH=/usr/local/cuda-8.0/bin:$PATH
source ~/.bashrc

安装tensorflow

export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0rc1-cp27-none-linux_x86_64.whl

pip install --upgrade $TF_BINARY_URL