Technic

教你在百度AI Studio下安装tensorflow和Keras

2020-05-20

(说在前边,AI Studio真是个好东西)

最近开始使用百度AI Studio,由于对飞桨现在还不熟悉,所以想先跑一下基于keras的模型。不过AI Studio上没有预装keras和tensorflow,得自己装一下。

有两个背景原因先介绍一下:

  1. 在notebook中使用pip命令安装的库在环境下次启动时候就没有了。
  2. 现在已经没法在notebook中安装tensorflow了,不被允许,所以tf的安装必须在终端里进行。

现在说一下解决方案,以便能够安装和长期使用自己安装的库:

  1. 建立一个external-libraries的文件夹
  2. 将keras和tensorflow-gpu用pip方式安装并定位在这个文件夹中
  3. 之后运行程序的时候,在notebook中使用sys添加之前的文件夹到path中
  4. (注:此方法也适用于其他库)

具体按步骤放一下代码,对于步骤1和2,如果是在终端中操作,用直接复制代码即可;如果是在notebook中,需要在代码前添加叹号“!”(tf的安装必须在终端里进行):

  1. 建立文件夹 mkdir /home/aistudio/external-libraries
  2. 安装keras和tensorflow-gpu,把PACKAGE_NAME换为对应的库的名字即可 pip install PACKAGE_NAME -t /home/aistudio/external-libraries
  3. 在Notebook开头加入以下代码即可
    import sys
    sys.path.append('/home/aistudio/external-libraries')

其他需要说明的问题:

  1. 感觉在AI Studio里使用python/python3,pip/pip3没啥区别;
  2. 我在安装keras和tensorflow的时候提示# ERROR: paddlepaddle 1.7.1 has requirement scipy<=1.3.1; python_version >= "3.5", but you'll have scipy 1.4.1 which is incompatible.不过并没有对结果产生什么影响,因为后续会用keras和tf,所以这个ERROR应该问题不大;
  3. 我参考的网页里还提到了以下错误,不过我没有遇到TypeError: init() got an unexpected keyword argument ‘serialized_options’。解决方法是使用!pip install -U protobuf

最后上个图吧,有图有真相,实测环境重启之后也可以使用:

参考资料:AI Studio下如何跑tensorflow

Leave a Reply

Your email address will not be published. Required fields are marked *