温馨提示×

怎样查ubuntu pytorch版

小樊
52
2026-10-19 11:35:36
栏目: 智能运维

在 Ubuntu 上查看已安装的 PyTorch 版本,常用方法如下:

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

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

或进入 Python 交互环境:

python3
>>> import torch
>>> print(torch.__version__)

如果你用的是 conda 环境,先激活环境再执行:

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

方法二:查看是否为 CUDA 版本

python3 -c "import torch; print(torch.version.cuda)"
  • 输出 None:CPU 版本
  • 输出类似 11.8:CUDA 版本

还可以看是否能用 GPU:

python3 -c "import torch; print(torch.cuda.is_available())"

方法三:用 pip 查看

pip3 show torch

或

pip3 list | grep torch

如果是 conda 安装的:

conda list | grep pytorch

方法四:查看安装来源(pip / conda)

pip3 show torch | grep Location

小提示

  • Ubuntu 本身不“自带” PyTorch
  • PyTorch 版本和 Python 版本、CUDA 驱动有关
  • 推荐用虚拟环境(venv / conda)管理

如果你愿意,可以告诉我:

  • 是 pip 还是 conda 安装
  • 是否用 GPU / CUDA 我可以帮你判断版本是否合理或如何升级。

0 踩