如何通过诗歌安装最新的Pytorch预览（夜间）构建

Question

最新的预览构建支持在M1 MacBook Pros上进行了加速培训。目前，此功能仅由最新的预览（夜间）构建支持：

要开始，只需安装最新预览（夜间） 您的Apple Silicon Mac运行MacOS 12.3或更高版本 Python的版本（ARM64）。

根据文档，这就是如何通过 pip /code>：

pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu

但是我正在使用诗歌来管理我的Python项目依赖项。

诗歌似乎并不支持 - pre 选项：

$ poetry add --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu 

  Stack trace:

  11  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/console_application.py:123 in run
      io = io_factory(

  10  ~/.poetry/lib/poetry/console/config/application_config.py:221 in create_io
      resolved_command = application.resolve_command(args)

   9  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/console_application.py:110 in resolve_command
      return self._config.command_resolver.resolve(args, self)

   8  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/default_resolver.py:34 in resolve
      return self.create_resolved_command(result)

   7  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/default_resolver.py:166 in create_resolved_command
      if not result.is_parsable():

   6  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/resolve_result.py:43 in is_parsable
      self._parse()

   5  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/resolve_result.py:49 in _parse
      self._parsed_args = self._command.parse(self._raw_args)

   4  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/api/command/command.py:113 in parse
      return self._config.args_parser.parse(args, self._args_format, lenient)

   3  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:53 in parse
      self._parse(args, _fmt, lenient)

   2  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:101 in _parse
      self._parse_long_option(token, tokens, fmt, lenient)

   1  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:247 in _parse_long_option
      self._add_long_option(name, None, tokens, fmt, lenient)

  NoSuchOptionException

  The "--pre" option does not exist.

  at ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:300 in _add_long_option
      296│     def _add_long_option(
      297│         self, name, value, tokens, fmt, lenient
      298│     ):  # type: (str, Optional[str], List[str], ArgsFormat, bool) -> None
      299│         if not fmt.has_option(name):
    → 300│             raise NoSuchOptionException(name)
      301│ 
      302│         option = fmt.get_option(name)
      303│ 
      304│         if value is False:

如何使用诗歌添加最新的预览（每晚）Pytorch的构建？

Answer 1

您的 pip 命令的诗相等：

poetry source add --priority explicit pytorch https://download.pytorch.org/whl/nightly/cpu

然后：

poetry add --source pytorch torch torchvision

第一个命令添加了包装的新源URL（默认情况下，只有PYPI）， explicit> explicit 优先级表示如果未在使用诗歌添加时明确指定源头，则使用PYPI。第二个命令从指定的源添加 torch 和 torchvision 。

但是，由于依赖性解决问题，我无法通过这种平均每晚安装Pytorch。

有第二个解决方案，但是它要求您手动为建筑和环境选择分配的轮子。

首先，请访问并找到近期匹配您的架构和环境的车轮。例如，如果您在Apple Silicon上使用Python 3.10，则将是 torch-2.1.0.dev20230417-CP310-NONE-MACOSX_11_11_0_ARM64.WHL 。注意 CP310 ， MACOSX 和 ARM64 零件。

完成第一步后，请重复使用Torchvision。然后，您可以使用诗歌添加添加这些依赖项：

poetry add "https://download.pytorch.org/whl/nightly/cpu/torch-2.1.0.dev20230416-cp310-none-macosx_11_0_arm64.whl" "https://download.pytorch.org/whl/nightly/cpu/torchvision-0.16.0.dev20230416-cp310-cp310-macosx_11_0_arm64.whl"

或直接在 pyproject.toml 中：

[tool.poetry.dependencies]
torch = {url = "https://download.pytorch.org/whl/nightly/cpu/torch-2.1.0.dev20230416-cp310-none-macosx_11_0_arm64.whl"}
torchvision = {url = "https://download.pytorch.org/whl/nightly/cpu/torchvision-0.16.0.dev20230416-cp310-cp310-macosx_11_0_arm64.whl"}

Answer 2

我通过在 https://docs.conda.io.io.io.io.io.io /en/latest/miniconda.html （M1 64位）