AI Studio上安装TensorFlow GPU版本

本文详细介绍了在百度的AIStudio平台上安装TensorFlow GPU版本的步骤,包括使用国内镜像下载、创建文件夹存放库文件及升级protobuf版本等关键操作。

AI Studio是百度开发的基于其深度学习框架paddlepaddle(飞桨)的AI一站式平台,通过为大家提供免费的顶级GPU(Tesla v100)来推广飞桨框架的,整体感觉paddle还是挺好用的,废话不多说了,接下来看下我们如何在AI studio上安装TensorFlow的GPU版本吧!

 

Step1:打开部署的环境进入终端。

Step2:新建一个文件夹来保存我们要下载的TensorFlow包。

mkdir /home/aistudio/external-libraries

Step3:下载TensorFlow的GPU版本,记住要用国内的镜像

pip install tensorflow-gpu -t /home/aistudio/external-libraries -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

如果这样在python3下引入TensorFlow还是会出错的

Step4:升级protobuf版本,这个服务器上的版本是3.1,要升级到3.9

pip install -U protobuf -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com

到现在为止才是真正的实现了在AI studio上安装TensorFlow的GPU版本了。

那么只需要在终端进入python环境即可了。下面就可以打印TensorFlow的相关信息了

import sys
sys.path.append('/home/aistudio/external-libraries')
import tensorflow as tf
print(tf.__version__)

 

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值