05月17, 2019

Tensorflow-gpu在Ubuntu 上的安装步骤

在ubuntu手工安装tensorflow步骤较繁琐,下面是安装步骤

1. 安装tensorflow-gpu==1.13.1

首先我要安装的tensorflow版本是1.13.1,它对应的CUDA版本必须是10.0。

pip3 install tensorflow-gpu==1.13.1

2. 安装nvidia driver

如果已经安装了nvidia-driver,需要御载重新安装

# 清除已经安装的驱动
dpkg purge nvidia*
dpkg purge libnvidia*

# 安装410驱动,对应cuda 10.0
apt install nvidia-driver-410

# 重启
reboot

3. 安装cuda10.0

需要到nvidia的官网下载

4. 安装cudnn7.5

由于cudnn是nvidia的版权,必须先注册帐号,登陆后才能下载

常见错误

1) cudnn没有安装

错误消息:

ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory

PyPI使用国内源

通过几次 pip 的使用,对于默认的 pip 源的速度实在无法忍受,于是便搜集了一些国内的pip源,如下:

阿里云 http://mirrors.aliyun.com/pypi/simple/

中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/

豆瓣(douban) http://pypi.douban.com/simple/

清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/

中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

本文链接:http://57km.cc/post/install tensorflow-gpu on ubuntu.html

-- EOF --

Comments