深度学习Apex库出现报错‘IndexError: tuple index out of range‘的解决方法

bug产生原因：使用了高版本的python或者apex库【到底是哪一个还不确定】，导致GPU上运行的pytorch深度学习代码出现’IndexError: tuple index out of range’报错。

解决方法：
从github上https://github.com/NVIDIA/apex下载的apex库在正式安装之前，首先修改源代码。
修改源代码中的apex-master/apex/amp/utils.py文件
将

if cached_x.grad_fn.next_functions[1][0].variable is not x:
1

修改为

if cached_x.grad_fn.next_functions[0][0].variable is not x:
1

之后再进入apex-master文件夹执行

pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --global-option="--cpp_ext" --global-option="--cuda_ext" ./
1

编译安装。
实测这个解决方案成功修复了报错的问题。