如何解决在Google Colab中安装TA-Lib时的numpy相关报错？

问题背景 #

此前用于Colab安装TA-Lib的代码现已失效，执行后抛出ImportError: numpy.core.multiarray failed to import错误，原安装代码如下：

!wget http://prdownloads.sourceforge.net/ta-lib/ta-lib-0.4.0-src.tar.gz
!tar -xzvf ta-lib-0.4.0-src.tar.gz
%cd ta-lib/
!./configure --prefix=/usr
!make
!make install
!pip install Ta-Lib
import talib

报错原因 #

Colab默认numpy版本更新后，与编译TA-Lib时依赖的numpy版本不兼容，导致导入时底层数组模块加载失败。

解决步骤 #

• 卸载现有冲突依赖：

  !pip uninstall -y ta-lib numpy
  

• 安装与TA-Lib兼容的numpy版本：

  !pip install numpy==1.21.6
  

• 重新编译并安装TA-Lib：

  !wget http://prdownloads.sourceforge.net/ta-lib/ta-lib-0.4.0-src.tar.gz
  !tar -xzvf ta-lib-0.4.0-src.tar.gz
  %cd ta-lib/
  !./configure --prefix=/usr
  !make
  !make install
  !pip install Ta-Lib
  

• 验证导入是否成功：

  import talib
  print("TA-Lib导入成功")
  

额外排查方案 #

如果执行后仍报错，可先清理编译残留文件再重试：

%cd /content
!rm -rf ta-lib ta-lib-0.4.0-src.tar.gz

内容的提问来源于stack exchange，提问作者Christos Spanos