Ubuntu下TensorFlow-GPU安装

2020-04-06  179 次阅读


本文最后更新于2020年4月13日,已超过 5 个月没更新!

安装Anaconda

进入官网下载,之后终端输入:

$ bash Anaconda3-2020.02-Linux-x86_64.sh

一路enter, yes下去,就完成了。
之后重启终端可以执行以下命令进行验证。

$ conda list

安装tensorflow-gpu

  1. 以下命令需要科学上网走代理
$ pip --default-timeout=1000000 install  tensorflow-gpu

若出现代理错误,则执行如下命令后再进行安装。

$ unset all_proxy && unset ALL_PROXY
$ export all_proxy="socks5://127.0.0.1:1080"
  1. 建议使用国内镜像进行安装。
$ pip --default-timeout=1000000 install -U -i https://pypi.tuna.tsinghua.edu.cn/simple/ --upgrade tensorflow-gpu

此处说明一下命令参数:

  • -i: 指定库的安装源
  • -U:升级原来已经安装的包或安装更新到最新版本。

其他镜像网址:

  • 清华:https://pypi.tuna.tsinghua.edu.cn/simple
  • 阿里云:http://mirrors.aliyun.com/pypi/simple/
  • 中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
  • 豆瓣:http://pypi.douban.com/simple/

配置CUDA和cuDNN

下表为TensorFlow1.14-2.10版本所对应的CUDA和cuDNN版本。

TensorFlow版本 CUDA版本 cuDNN版本
1.14 CUDA Toolkit 10.0 cuDNN v7.4
1.15 CUDA Toolkit 10.0 cuDNN v7.5
2.0 CUDA Toolkit 10.0 cuDNN v7.6
2.1.0 CUDA Toolkit 10.1 cuDNN v7.6

CUDA

下载CUDA安装包后,执行以下命令:

$ sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
$ sudo apt-key add /cuda-repo-9-0-local/7fa2af80.pub
$ sudo apt-get update
$ sudo aptitude install cuda

终端输入以下命令查看是否安装成功。

$ nvcc --version

cuDNN

下载cnDNN安装文件解压后,执行以下命令:

$ 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
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

终端输入以下命令查看是否安装成功。

$ cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

参考链接

ubuntu下安装cuda+cudnn步骤
Ubuntu安装和卸载CUDA和CUDNN


Try and fail, but don't fail to try.