我使用以下命令pip安装nvidia-tensorflow[horovod]时出现错误。

Question

(tensorholo) C:\Users\alaba\Desktop\MIT PROJECT\tensor_holography-main>pip install nvidia-tensorflow[horovod]
Collecting nvidia-tensorflow[horovod]
  Downloading nvidia-tensorflow-0.0.1.dev5.tar.gz`` (7.9 kB)
  Preparing metadata (setup.py) ... error
  error: subprocess-exited-with-error


  × python setup.py egg_info did not run successfully.
  │ exit code: 1
  ╰─> [19 lines of output]
      Traceback (most recent call last):
        File "<string>", line 2, in <module>
        File "<pip-setuptools-caller>", line 34, in <module>
        File "C:\Users\alaba\AppData\Local\Temp\pip-install-ztjzt002\nvidia-tensorflow_001bf7a090024c37a3c7ff91374ac7c4\setup.py", line 150, in <module>
          raise RuntimeError(open("ERROR.txt", "r").read())
      RuntimeError:
      ###########################################################################################
      The package you are trying to install is only a placeholder project on PyPI.org repository.
      This package is hosted on NVIDIA Python Package Index.

      This package can be installed as:
      ```
      $ pip install nvidia-pyindex
      $ pip install nvidia-tensorflow
      ```

      Please refer to NVIDIA instructions: https://github.com/NVIDIA/tensorflow#install.
      ###########################################################################################

此错误起源于子进程，很可能与pip无关。

error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

注:这是上述软件包的一个问题，而不是pip。提示:详见上文。

Answer 1

nvidia-tensorflow[horovod]包不支持Windows。您需要在WSL2 ( 2)中运行它，才能使它在Windows上工作。

就像官方医生说

nvidia-tensorflow包包括对Linux的CPU和GPU支持。

以及同一存储库上的本期：