获取火炬版本 - Python

如果你在使用PyTorch，获取PyTorch的版本信息可能是非常重要的。这可以帮助你确定你是否需要更新到最新版本，以及确定你是否正在使用正确的文档。在Python中获取PyTorch版本信息很容易，下面是几种不同的方法：

方法一：使用torch.version

PyTorch提供了一个版本信息模块torch.version，通过这个模块可以获取PyTorch的版本信息。

import torch

print(torch.__version__)

输出结果如下：

'1.9.0'

方法二：使用pip

另一种获取PyTorch版本的方法是使用pip。这种方法对于你正在使用的任何Python包都是通用的，不仅适用于PyTorch。

!pip show torch

输出结果如下：

Name: torch
Version: 1.9.0+cu102
Summary: Tensors and Dynamic neural networks in Python with strong GPU acceleration
Home-page: https://pytorch.org/
Author: PyTorch Team
Author-email: packages@pytorch.org
License: BSD-3-Clause
Location: /usr/local/lib/python3.7/dist-packages
Requires: typing-extensions, numpy, torch
Required-by: torchvision, torchaudio, kapre

方法三：使用conda

如果你使用的是conda包管理器，你可以通过以下命令来获取PyTorch版本信息：

!conda list | grep torch

输出结果如下：

torch                     1.9.0+cu102      pytorch

总结

以上就是获取PyTorch版本的三种不同方法。使用这些方法之一可以让你更好地了解你的PyTorch版本，以便在需要时更新到最新版本。