在 CentOS 上“查看 PyTorch”通常指以下几种情况之一,你可以根据需求选择对应方法:
pip show torch
或
pip list | grep torch
如果是 Python3:
pip3 show torch
conda list | grep torch
进入 Python 环境:
python
或
python3
然后执行:
import torch
print(torch.__version__)
import torch
print(torch.cuda.is_available()) # True 表示支持 GPU
print(torch.version.cuda) # CUDA 版本
print(torch.backends.cudnn.version()) # cuDNN 版本
import torch
print(torch.__file__)
nvidia-smi
(需要安装 NVIDIA 驱动)
查看 CUDA 工具包:
nvcc --version
No module named torch说明未安装 PyTorch,可使用:
pip install torch torchvision torchaudio
或根据官网选择版本: https://pytorch.org/get-started/locally/
常见错误与 libgomp、GLIBC 有关,可使用:
yum install libgomp
如果你能说明你是 查看版本 / 是否安装 / GPU 支持 / 报错信息,我可以给你更精确的命令。