在CentOS上安装PyTorch及其依赖项,可以按照以下步骤进行:
-
更新系统:
首先,确保你的CentOS系统是最新的。打开终端并运行以下命令来更新系统:sudo yum update -y
-
安装必要的依赖项:
PyTorch需要一些基本的依赖项,包括编译工具和库。你可以通过以下命令安装这些依赖项:sudo yum groupinstall -y "Development Tools" sudo yum install -y cmake3 git wget sudo yum install -y python3 python3-devel sudo yum install -y numpy
-
创建虚拟环境(可选):
为了避免与系统中的其他Python包发生冲突,建议创建一个虚拟环境:sudo yum install -y python3-virtualenv virtualenv pytorch_env source pytorch_env/bin/activate
-
安装PyTorch:
PyTorch提供了多种安装方式,包括通过pip安装预编译的包或从源代码编译。以下是通过pip安装预编译包的方法:-
CPU版本:
pip install torch torchvision torchaudio
-
GPU版本(需要CUDA支持):
首先,确保你的系统上已经安装了CUDA和cuDNN。然后,根据你的CUDA版本选择合适的PyTorch版本。例如,如果你的CUDA版本是11.7,可以使用以下命令:pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
-
-
验证安装:
安装完成后,可以通过以下命令验证PyTorch是否安装成功:import torch print(torch.__version__) print(torch.cuda.is_available()) # 如果安装了GPU版本,应该返回True
通过以上步骤,你应该能够在CentOS上成功安装PyTorch及其依赖项。如果在安装过程中遇到任何问题,请参考PyTorch官方文档或社区论坛寻求帮助。