pip安装nvidia-tensorflow[horovod]报错如何解决

PyPI公共仓库上架的nvidia-tensorflow是空占位包，无任何实际安装内容，所有正式发布版本（含horovod扩展版本）仅托管在NVIDIA自有的Python包源中，直接从默认公共源安装必然触发对应报错。

全程保持当前处于tensorholo虚拟环境中（即命令行前缀显示(tensorholo)的状态），按顺序执行命令：

1. 安装NVIDIA官方包索引工具，自动将NVIDIA私有包源添加到本地pip配置中

   pip install nvidia-pyindex
   

2. 上一步执行完成无报错后，执行带horovod扩展的安装命令即可

   pip install nvidia-tensorflow[horovod]
   

   执行后pip会自动从NVIDIA官方源拉取完整安装包，不会再下载公共源的占位空包。

• 目前nvidia-tensorflow最高支持Python 3.8版本，若虚拟环境Python版本高于3.8，会出现找不到对应安装包的问题，需先重建虚拟环境切换到兼容版本
• 安装过程中如果出现horovod相关编译报错，需提前在系统中安装好与CUDA版本匹配的cuDNN、NCCL组件后再重试
• 安装时不要手动指定清华源、阿里源这类第三方公共镜像源参数，这类镜像未同步NVIDIA私有源内容，强制使用仍会拉到占位包触发报错
• 若之前安装失败留下了错误缓存，可在安装命令中加--no-cache-dir参数跳过本地缓存，对应命令为pip install --no-cache-dir nvidia-tensorflow[horovod]

内容的提问来源于stack exchange，提问作者Ala BR