ubuntu

Ubuntu中如何查看PyTorch版本

小樊
49
2025-10-21 03:12:51
栏目: 智能运维

在Ubuntu系统中,你可以通过以下几种方法来查看PyTorch的版本:

方法一:使用Python解释器

  1. 打开终端(Terminal)。
  2. 输入 pythonpython3 启动Python解释器。
  3. 在Python解释器中,输入以下命令并按回车键:
import torch
print(torch.__version__)

这将输出当前安装的PyTorch版本。

方法二:使用命令行工具

  1. 打开终端(Terminal)。
  2. 输入以下命令并按回车键:
python -c "import torch; print(torch.__version__)"

或者,如果你使用的是Python 3,可以使用:

python3 -c "import torch; print(torch.__version__)"

这将直接在终端中输出PyTorch的版本号。

方法三:查看安装日志或文档

如果你是通过pip或conda安装的PyTorch,你可以查看安装日志或相关文档来确认版本信息。

pip show torch

在输出的信息中,你可以找到版本号。

conda list torch

在输出的信息中,你也可以找到版本号。

注意事项

pip install torch torchvision torchaudio

或者,如果你使用的是conda:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

通过以上方法,你应该能够轻松地在Ubuntu系统中查看PyTorch的版本。

0
看了该问题的人还看了