如何通过 Poetry 安装最新的 PyTorch 预览版(每晚)

Lei*_*eif 7 python pip package-management pytorch python-poetry

最新的预览版支持在 M1 MacBook Pro 上进行加速训练。目前仅最新预览版(每晚)版本支持此功能:

\n
\n

首先,只需在运行 macOS 12.3 或更高版本以及原生 Python 版本 (arm64) 的 Apple Silicon Mac 上安装最新的预览版(每晚)版本即可。

\n
\n

根据文档,这是通过以下方式安装最新预览版本的方法pip

\n
pip3 install --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu\n
Run Code Online (Sandbox Code Playgroud)\n

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

\n

诗歌似乎不支持该--pre选项:

\n
$ poetry add --pre torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/nightly/cpu \n\n  Stack trace:\n\n  11  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/console_application.py:123 in run\n      io = io_factory(\n\n  10  ~/.poetry/lib/poetry/console/config/application_config.py:221 in create_io\n      resolved_command = application.resolve_command(args)\n\n   9  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/console_application.py:110 in resolve_command\n      return self._config.command_resolver.resolve(args, self)\n\n   8  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/default_resolver.py:34 in resolve\n      return self.create_resolved_command(result)\n\n   7  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/default_resolver.py:166 in create_resolved_command\n      if not result.is_parsable():\n\n   6  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/resolve_result.py:43 in is_parsable\n      self._parse()\n\n   5  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/resolver/resolve_result.py:49 in _parse\n      self._parsed_args = self._command.parse(self._raw_args)\n\n   4  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/api/command/command.py:113 in parse\n      return self._config.args_parser.parse(args, self._args_format, lenient)\n\n   3  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:53 in parse\n      self._parse(args, _fmt, lenient)\n\n   2  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:101 in _parse\n      self._parse_long_option(token, tokens, fmt, lenient)\n\n   1  ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:247 in _parse_long_option\n      self._add_long_option(name, None, tokens, fmt, lenient)\n\n  NoSuchOptionException\n\n  The "--pre" option does not exist.\n\n  at ~/.poetry/lib/poetry/_vendor/py3.9/clikit/args/default_args_parser.py:300 in _add_long_option\n      296\xe2\x94\x82     def _add_long_option(\n      297\xe2\x94\x82         self, name, value, tokens, fmt, lenient\n      298\xe2\x94\x82     ):  # type: (str, Optional[str], List[str], ArgsFormat, bool) -> None\n      299\xe2\x94\x82         if not fmt.has_option(name):\n    \xe2\x86\x92 300\xe2\x94\x82             raise NoSuchOptionException(name)\n      301\xe2\x94\x82 \n      302\xe2\x94\x82         option = fmt.get_option(name)\n      303\xe2\x94\x82 \n      304\xe2\x94\x82         if value is False:\n
Run Code Online (Sandbox Code Playgroud)\n

如何使用 Poetry 添加 PyTorch 的最新预览(每晚)版本?

\n

小智 -1

我通过在https://docs.conda.io/en/latest/miniconda.html(M1 64 位)为 m1 arm 安装 minicoda 来使其工作