Ubuntu 18.04安装Tensorflow(GPU)

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

查看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是否安装成功的验证。

文章来源网络,作者:管理,如若转载,请注明出处:https://shuyeidc.com/wp/201037.html<

(0)
管理的头像管理
上一篇2025-04-05 14:01
下一篇 2025-04-05 14:02

相关推荐

  • 云服务器和云虚拟主机怎么选?云服务器和虚拟主机区别

    云服务器适合业务增长快、需弹性扩展的场景,而云虚拟主机适合预算有限、技术门槛低的小型静态网站或测试环境,二者核心区别在于资源独享性与运维复杂度,核心差异解析:从底层架构到使用体验很多人容易混淆这两者,觉得它们都是“买空间建站”,它们的底层逻辑完全不同,云服务器(ECS)就像是你租了一整栋别墅,水电网络独立,你想……

    2026-06-29
    0
  • 赣州智慧旅游招聘是真的吗?赣州旅游人才招聘信息

    中级岗位(3-5年经验)月薪范围通常在6000-10000元,这类岗位需要独立负责项目模块,如独立运营一个抖音账号,或维护一个景区小程序的功能迭代,具备成功案例的候选人议价能力较强,高级岗位(5年以上经验)月薪范围通常在10000-20000元,部分核心管理岗可达更高,这类人才需要具备战略规划能力,如制定整个景……

    2026-06-29
    0
  • 赣州智能物联网车位锁如何管理?智能车位锁管理系统多少钱

    赣州智能物联网车位锁管理的核心在于通过云端平台实现远程控锁、状态实时监控及自动计费,彻底解决传统车位“被占难管”与“找位难”的痛点,在赣州这样的城市,随着机动车保有量的持续增长,老旧小区、商业综合体以及私人固定车位的资源矛盾日益凸显,传统的机械地锁或简易遥控锁,不仅操作繁琐,更无法实现数据化管理,引入智能物联网……

    2026-06-29
    0
  • 赣州智能消防栓好用吗,智能消防栓多少钱一个

    赣州智能消防栓通过物联网技术实现实时监测与远程报警,能显著降低火灾响应时间并提升城市消防安全管理水平,是目前智慧城市建设中不可或缺的基础设施,赣州智能消防栓的核心价值与应用场景传统消防栓往往存在“看不见、摸不着、用不了”的痛点,在赣州这样地形复杂、老城区与新城区并存的区域,传统设施的管理难度极大,智能消防栓的出……

    2026-06-29
    0
  • 云服务器和物理机到底有啥区别?

    云服务器本质上是虚拟化资源池中的弹性实例,而传统物理服务器是独占的硬件实体,前者胜在弹性与运维便捷,后者强在物理隔离与性能稳定,具体选择取决于业务对成本、扩展性及安全合规的权衡,很多人初次接触服务器时,容易把“云服务器”和“传统物理服务器”混为一谈,觉得它们都是用来跑网站或存数据的盒子,这两者的底层逻辑完全不同……

    2026-06-29
    0

发表回复

您的邮箱地址不会被公开。必填项已用 * 标注