问题描述
安装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
即可安装文章来源:https://www.toymoban.com/news/detail-415790.html
注意!!
在安装过程中许多依赖项也是需要从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/下载即可文章来源地址https://www.toymoban.com/news/detail-415790.html
到了这里,关于安装nvidia-tensorflow时出现Preparing metadata (setup.py) ... error的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!