在 Linux 下查看 PyTorch 版本,常用有以下几种方法:
python -c "import torch; print(torch.__version__)"
或先进入 Python 交互环境:
import torch
print(torch.__version__)
如果有 CUDA 支持,还可以查看 CUDA 版本:
print(torch.version.cuda)
pip show torch
或
pip list | grep torch
conda list torch
import torch
print(torch.cuda.is_available()) # True 表示支持 CUDA
如果你不确定用的是哪个 Python 环境,建议先确认:
which python
which pip
避免查看错环境。
如果你是 源码编译的 PyTorch,版本可能显示为类似 2.1.0+a0c3b88。需要我帮你判断 CPU/GPU 版本或升级方法也可以。