📜  在 ubuntu 中获取 pytorch 版本版本 - Shell-Bash (1)

📅  最后修改于: 2023-12-03 15:23:20.667000             🧑  作者: Mango

在Ubuntu中获取PyTorch版本

如果你正在使用Ubuntu操作系统,并想要找出你的PyTorch版本,以下是一些方法:

方法1:使用命令行

打开终端,输入以下命令:

pip show torch

这将显示你当前安装的PyTorch版本信息,如下所示:

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.8/dist-packages
Requires: numpy, typing-extensions, torch
Required-by: torchvision, fastai
方法2:使用Python代码

如果你更喜欢使用Python代码来获取PyTorch版本,可以使用以下代码:

import torch

print(torch.__version__)

你将获得类似于以下输出:

1.9.0+cu102

这是安装的PyTorch版本。

我们希望这篇文章能够帮助你在Ubuntu中快速获取PyTorch版本信息。