在CentOS上安装PyTorch通常涉及以下步骤:
更新系统: 首先,确保你的CentOS系统是最新的。你可以使用以下命令来更新系统:
sudo yum update -y
安装Miniconda: Miniconda是一个轻量级的Anaconda发行版,包含了conda、Python以及180多个科学包及其依赖项。你可以从Miniconda的官方网站下载适合你系统的安装包,并按照安装向导进行安装。
下载链接:https://docs.conda.io/en/latest/miniconda.html
安装示例(以Linux版为例):
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh
安装过程中,你需要接受许可协议、选择安装路径等。完成后,重启你的shell或终端。
激活conda环境: 安装完成后,你需要激活conda环境。你可以创建一个新的环境并激活它,或者直接使用base环境。
创建新环境并激活:
conda create -n myenv python=3.8
conda activate myenv
激活base环境(如果你不想创建新环境):
conda activate base
安装PyTorch: 在激活的环境中,你可以使用conda或pip来安装PyTorch。由于CentOS可能没有官方的PyTorch包,你可能需要添加PyTorch的官方channel。
使用conda安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch -c conda-forge
注意:cudatoolkit=11.3
中的版本号可能需要根据你的CUDA版本进行调整。你可以通过运行conda info cudatoolkit
来查看可用的CUDA版本。
使用pip安装(如果conda安装失败或不可用):
pip install torch torchvision torchaudio
验证安装: 安装完成后,你可以验证PyTorch是否安装成功。运行以下Python代码:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
如果一切正常,你应该能看到PyTorch的版本号以及CUDA是否可用(取决于你的系统配置)。
请注意,以上步骤可能会因系统版本、CUDA版本和PyTorch版本的不同而有所变化。建议查阅PyTorch的官方文档以获取最准确的安装指南。