linuxsir首页 LinuxSir.Org | Linux、BSD、Solaris、Unix | 开源传万世,因有我参与欢迎您!
网站首页 | 设为首页 | 加入收藏
您所在的位置:主页 > Linux基础建设 >

Ubuntu 18.04安装Tensorflow(GPU)

时间:2019-06-11  来源:未知  作者:admin666

1.Ubuntu安装Python3.6:

首先拉取远程仓库

sudo add-apt-repository ppa:jonathonf/python-3.6 

更新源

sudo apt-get update

在线安装

sudo apt-get install python3.6

补充,其实安装python3用一句:sudo apt-get install python3-dev 即可。查看版本:python3 --version

更改默认值,python默认为Python2,现在修改为Python3

sudo update-alternatives --install /usr/bin/python python /usr/bin/python2 100

sudo update-alternatives --install /usr/bin/python python /usr/bin/python3 150

2.Ubuntu安装nvidia显卡驱动(参考上一篇博客)

查看nvidia显卡驱动版本

nvidia-smi  

3.安装对应版本的CUDA

在安装CUDA时一定要注意其与英伟达显卡驱动以及Linux系统和GCC版本的对应关系,如果版本之间不匹配,是安装不成功的。

CUDA与Driver的对应版本
参考链接:https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
cuda8.0与Linux系统以及GCC的对应关系
参考链接:https://docs.nvidia.com/cuda/archive/8.0/cuda-installation-guide-linux/index.html
cuda9.0与Linux系统以及GCC的对应关系
参考链接:https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html
cuda10.0与Linux系统以及GCC的对应关系
参考链接:https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

查看gcc版本:

gcc -v

由于RTX2060显卡驱动在Ubuntu18.04上是418.56版本,所以选择对应的cuda10.1版本安装

cuda官网下载:https://developer.nvidia.com/cuda-toolkit

另附历史版本:https://developer.nvidia.com/cuda-toolkit-archive

CUDA安装命令:
sudo sh cuda_xxx_linux.run

4.安装CUDA对应的CUDNN版本

cudnn官网下载(需要注册):https://developer.nvidia.com/cudnn

另附:https://developer.nvidia.com/rdp/cudnn-archive 

下载Ubuntu18.04对应的CUDNN安装包,然后进入CUDNN安装包所在目录,执行以下命令:

sudo dpkg -i runtime包.deb 
sudo dpkg -i developer包.deb 
sudo dpkg -i 代码sample包.deb

至此,CUDNN安装完成。

5.安装对应版本的Tensorflow

首先要清楚最新版Tensorflow最多支持到CUDA哪个版本?

https://www.tensorflow.org/install/install_sources#common_installation_problems

安装pip3(针对python3): 

sudo apt-get install python3-pip

官方推荐是用Virtualenv安装,不过这里我们仅使用pip进行安装。

sudo pip3 install tensorflow-gpu

 我现在这里安装的是tensorflow_gpu-1.13.1。由于CUDA最新版本是10.1,但是目前最新的tensorflow1.13.1还不支持这个版本,所以只能用CUDA10.0。

推荐搭配:CUDA10.0+CUDNN7.5+tensorflow-gpu1.13.1+python3.6.7

卸载cuda和cudnn,重新安装:

sudo apt remove cudnn*
sudo apt-get remove cuda*
sudo apt-get autoclean
然后在目录切换到/esr/local/下
cd /usr/local/
sudo rm -r cuda-10.1
搞定啦
可以重新安装其他版本啦

设置root用户密码:

sudo passwd root

以下是编辑 profile文件命令:
1.su 然后输入密码 进入root 2.gedit etc/profile 3.编辑保存.

首先确认/etc/profile中的路径包含了cuda10.0的安装路径及相应的库文件
编辑/etc/profile添加:

export PATH=$PATH:/usr/local/cuda-10.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.0/lib64

然后
source /etc/profile
使配置文件生效,再次执行。

安装vim:
sudo apt-get install vim
vim --version
问题报错:ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory
由于服务器TensorFlow经常报这个错误,
步骤1:
sudo vim ~/.bashrc
按下i进入编辑
在末位加入:
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.0
按下ESC退出编辑,输入:wq命令进行保存。
使之生效
source ~/.bashrc
步骤2:
据说在修改了步骤1就好了。但是每次我步骤1弄完,问题依旧存在。但是只需要步骤2,问题就可以解决。
检查 /usr/local/cuda-10.0/lib64 下是否有 libcublas.so.10.0
如果有,终端输入:
sudo ldconfig /usr/local/cuda-10.0/lib64

然后终端输入python

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

进行tensorflow是否安装成功的验证。

Ubuntu 18.04安装Tensorflow(CPU)  https://www.linuxidc.com/Linux/2019-06/158934.htm

Linux公社的RSS地址:https://www.linuxidc.com/rssFeed.aspx

友情链接
  • 适用于Windows 10的Microsoft Edge Dev 76.0.172.0已发布
  • 程序员人脸识别成人视频中的女性引发争议
  • 安装Oracle 11g RAC时报Failed to run “oifcfg” 和 找不到集群
  • Unity Tech正式宣布推出适用于Linux的新Unity编辑器
  • Oracle插入数据时出现 ORA
  • Phoronix Test Suite 8.8.1 发布
  • MySQL中使用group
  • X.Org Server 1.20.5 发布,一些XWayland和GLAMOR修复
  • MySQL中group by 与 order by 一起使用排序问题
  • Fedora 31考虑使用Zstd压缩其RPM包而不是XZ
  • PostgreSQL数据库安装部署
  • Google Chrome 成为互联网的看门人
  • Redis中Scan命令的使用
  • 华为Mate 20 Pro重回Android Q Beta支持名单中
  • TypeScript 3.5 发布,新增Omit 辅助类型
  • Redis集群模式下的redis
  • PyPI宣布2FA用于提高Python包下载安全性
  • MuseScore 3.1 发布,音乐制谱软件
  • openSUSE Leap 42.3 Linux操作系统将于2019年6月30日停止支持
  • Linux新闻 第974页
  • 美国司法部正计划对谷歌进行反垄断调查
  • 超六成Windows 10用户运行着一年前的版本
  • Linux里如何配置本地yum源和外网源
  • MySQL中group by 与 order by 一起使用排序问题
  • yum更换国内源及yum下载rpm包
  • MySQL中使用group_concat遇到的问题及解决
  • Linux 定时任务 Crontab
  • Oracle插入数据时出现 ORA-00001: unique constraint
  • Yii2 的安装及简单使用
  • SpringBoot配置加载顺序
  • git merge的使用
  • 编写shell脚本自动生成开头注释简介
  • JDK动态代理在RPC框架中的应用
  • 将Fedora 29升级到Fedora 30
  • Linux touch命令详述
  • Java实现对图片透明化处理
  • 如何在Ubuntu 18.04,19.04中安装uGet 2.2.2
  • ACL权限设置命令setfacl和getfacl命令
  • Nginx+Tomcat8+Memcached实现负载均衡及session共享
  • Redis集群模式下的redis-py-cluster方式读写测试
  • Linux命令 - facl文件系统的访问控制列表详解
  • 在Linux系统中创建SSH服务器别名
  • Redis中Scan命令的使用
  • Apache Camel XML外部实体注入漏洞(CVE-2019-0188)
  • shell脚本里使用echo输出颜色
  • PostgreSQL数据库安装部署
  • Apache Hadoop远程权限提升漏洞(CVE-2018-8029)
  • Google Chrome成为互联网的看门人
  • 欧洲杯直播频道
  • 彩票平台
  • 澳门银河国际网站
  • 28评测网
  • 新利棋牌
  • 中国娱乐场