AI Studio上安装TensorFlow GPU版本

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__)