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

3709

被折叠的 条评论
为什么被折叠?



