安装nvidia-tensorflow时出现Preparing metadata (setup.py) … error

问题描述

安装nvidia-tensorflow时执行语句

pip install nvidia-tensorflow[horovod]

出现如下错误:


这里的错误非常令人头疼,GitHub上nvidia官方论坛里的讨论贴也无法解决。

问题原因

出现这个bug的原因是我们采用了两个源进行package的下载
在执行语句pip install nvidia-tensorflow[horovod]前我们通常会执行

pip install nvidia-pyindex

这条语句会自动生成pip.conf,并加入额外源 https://pypi.ngc.nvidia.com
我们的安装进程在下载nvidia-tensorflow包时会首先在pip默认源https://pypi.org/simple中进行寻找
但找到的包是一个占位符(placeholder),只有名字没有内容,真正的包在nvidia官方服务器里,所以我们只需要去官方网址下载即可
nvidia-tensorflow下载链接

注意!

存在许多版本,每个nvidia-tensorflow都对应着不同的CUDA和cuDNN版本,可以在nvidia的GitHub官方处找到不同版本的信息
nvidia-tensorflow官方GitHub账号
在左上角可以查看历史分支

进入选中分支后可以在README.md文件的“Build From Source”中找到需求的CUDA和cuDNN版本,选择适合本机环境的版本下载即可
下载.whl文件后使用语句

pip install *****.whl

即可安装

注意!!

在安装过程中许多依赖项也是需要从nvidia服务器中下载的,因此可能会报错,如下所示

这时只需要将刚才下载nvidia-tensorflow安装包的链接中
https://developer.download.nvidia.com/compute/redist/nvidia-tensorflow/
的末尾改成我们需要下载的包即可
例如上图错误中我缺少nvidia-cuda-runtime-cu117包,只需要进入网址
https://developer.download.nvidia.com/compute/redist/nvidia-cuda-runtime-cu117/下载即可

物联沃分享整理
物联沃-IOTWORD物联网 » 安装nvidia-tensorflow时出现Preparing metadata (setup.py) … error

发表评论