centos怎么查看pytorch

centos

小樊

2026-09-09 10:39:45

栏目: 智能运维

在 CentOS 上查看 PyTorch 是否安装以及版本信息，一般按下面几种方式来：

一、查看 PyTorch 是否安装 & 版本

pip show torch

或

pip3 show torch

输出示例：

Name: torch
Version: 2.1.0

conda list torch

python

或

python3

然后输入：

import torch
print(torch.__version__)

查看 CUDA 是否可用：

print(torch.cuda.is_available())
print(torch.version.cuda)

import torch
print(torch.__file__)

nvcc --version

或

cat /usr/local/cuda/version.txt

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

✅ import torch 报错

✅ 多个 Python 环境 先用：

which python
which pip

确认环境一致

如果你愿意，可以告诉我：

我可以给你更精确的命令 ✅