温馨提示×

centos如何查看pytorch

小樊
62
2026-10-12 05:42:19
栏目: 智能运维

在 CentOS 上“查看 PyTorch”通常指以下几种情况之一,你可以根据需求选择对应方法:


一、查看是否已安装 PyTorch

1. 使用 pip 查看

pip show torch

或

pip list | grep torch

如果是 Python3:

pip3 show torch

2. 使用 conda 查看(如果你用 Anaconda)

conda list | grep torch

二、查看 PyTorch 版本

进入 Python 环境:

python

或

python3

然后执行:

import torch
print(torch.__version__)

三、查看 PyTorch 是否支持 CUDA(GPU)

import torch
print(torch.cuda.is_available())   # True 表示支持 GPU
print(torch.version.cuda)          # CUDA 版本
print(torch.backends.cudnn.version())  # cuDNN 版本

四、查看 PyTorch 安装路径

import torch
print(torch.__file__)

五、查看系统 CUDA 驱动(CentOS 层面)

nvidia-smi

(需要安装 NVIDIA 驱动)

查看 CUDA 工具包:

nvcc --version

六、常见问题

1. 提示 No module named torch

说明未安装 PyTorch,可使用:

pip install torch torchvision torchaudio

或根据官网选择版本: https://pytorch.org/get-started/locally/

2. CentOS 缺少依赖

常见错误与 libgomp、GLIBC 有关,可使用:

yum install libgomp

如果你能说明你是 查看版本 / 是否安装 / GPU 支持 / 报错信息,我可以给你更精确的命令。

0 踩