温馨提示×

linux pytorch如何查看版本

小樊
79
2026-09-25 17:34:55
栏目: 智能运维

在 Linux 下查看 PyTorch 版本,常用有以下几种方法:

方法一:在 Python 中查看(最常用)

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

或先进入 Python 交互环境:

import torch
print(torch.__version__)

如果有 CUDA 支持,还可以查看 CUDA 版本:

print(torch.version.cuda)

方法二:使用 pip 查看

pip show torch

或

pip list | grep torch

方法三:使用 conda 查看(如果你用 conda 环境)

conda list torch

方法四:查看是否是 GPU / ROCm 版本

import torch
print(torch.cuda.is_available())   # True 表示支持 CUDA

如果你不确定用的是哪个 Python 环境,建议先确认:

which python
which pip

避免查看错环境。

如果你是 源码编译的 PyTorch,版本可能显示为类似 2.1.0+a0c3b88。需要我帮你判断 CPU/GPU 版本或升级方法也可以。

0 踩