获取 PyTorch 版本 - Python

如果您正在使用 PyTorch 进行深度学习项目开发，那么您可能需要知道当前您的 PyTorch 版本是什么。以下是在 Python 中获取 PyTorch 版本的几种方法：

方法一：使用命令行

如果您正在使用 Linux 或 macOS 操作系统，您可以在终端中使用以下命令获取 PyTorch 版本：

pip show torch

此命令将输出 PyTorch 包的详细信息，包括版本号。例如：

Name: torch
Version: 1.6.0
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
...

方法二：使用代码

如果您正在编写 Python 脚本，您可以使用以下代码行获取 PyTorch 版本：

import torch

print(torch.__version__)

此代码将输出 PyTorch 的当前版本，例如：

1.6.0

方法三：使用环境变量

有时，您可能需要在代码中使用 PyTorch 的版本号，而不需要您的代码安装 PyTorch。在这种情况下，您可以使用以下代码行访问 PyTorch 版本号的环境变量：

import os

print(os.environ.get("TORCH_VERSION"))

如果您的代码正在运行的环境中安装了 PyTorch，这将输出 PyTorch 的版本号。否则，它将返回 None。

无论您选择哪种方法，都可以在开发过程中了解您的 PyTorch 版本。这有助于您查找有关文档或社区支持方面的信息，或确定当前代码是否与 PyTorch 的特定版本兼容。