ubuntu上安装tensorflow-gpu(前提是gpu驱动都配置好)

本文详细介绍了在Ubuntu系统中安装GPU版本TensorFlow的步骤,包括更新Python版本、安装pip、设置默认Python版本、安装依赖包及解决版本冲突等关键环节。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

ubuntu上安装tensorflow-gpu(前提是gpu驱动都配置好)

release版本安装

依赖环境:python

准备工作:

(1). 首先更新自己python版本为3.0+

安装python3
apt-get install python3

安装pip3
apt-get install python3-pip

为python3添加包
pip3 install packagename

安装pillow
apt-get install libjpeg-dev libfreetype6-dev zlib1g-dev libpng12-dev
pip3 install pillow

(2). 将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

正式安装:

(1). 安装pip:

sudo apt-get install python-pip python-dev   # for Python 2.7
sudo apt-get install python3-pip python3-dev # for Python 3.n

(2). 利用pip安装Tensorflow:

sudo pip install tensorflow-gpu

测试:

python
import tensorflow as tf

可能遇到的错误:

ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

这是tensorflow版本和cuda版本不匹配导致的,具体可查表格

解决方案:

如果你的cuda是8.0,那么你可以使用下面的指令将tensorflow的版本回滚到1.4版本

pip install --upgrade tensorflow-gpu==1.4

 

参考链接:https://blog.youkuaiyun.com/u014797226/article/details/80229887

                  https://www.jianshu.com/p/6f54b4f96556

                  https://blog.youkuaiyun.com/wangguchao/article/details/81776177 

                  https://www.cnblogs.com/yasmi/p/4832461.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值