使用pip安装nvidia-apex报错:apex error: class "at::Tensor" has no member "mutable_data_ptr
git下载官方repo之后,采用如下方法解决:
git checkout 2386a912164b0c5cfcd8be7a2b890fbac5607c82
然后安装
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./