温馨提示×

怎么查linux pytorch版本

小樊
74
2026-10-04 11:36:41
栏目: 智能运维

在 Linux 下查看已安装的 PyTorch 版本,常用以下几种方法:


方法一:通过 Python 代码查看(最常用)

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

如果你用的是 Python 3 且命令是 python3:

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

如果你使用的是虚拟环境(如 conda、venv),请先激活环境再执行。


方法二:进入 Python 交互式环境

python

然后输入:

import torch
print(torch.__version__)

退出按 Ctrl + D 或输入 exit()


方法三:查看 CUDA 支持的 PyTorch 版本

如果想确认 PyTorch 是否支持 CUDA:

import torch
print(torch.__version__)
print(torch.cuda.is_available())   # True 表示支持 CUDA
print(torch.version.cuda)          # CUDA 版本

方法四:通过 pip 查看

pip show torch

或:

pip list | grep torch

如果是 conda 环境:

conda list | grep torch

方法五:查看安装路径(辅助)

import torch
print(torch.__file__)

如果你不确定自己用的是哪种环境(系统 Python / conda / venv),可以先运行:

which python

确认当前 Python 解释器路径。

需要我帮你判断某个环境下的 PyTorch 是否正常吗?

0 踩