在 Linux 系统中安装 PyTorch 的方法如下:
- 首先,确保您的系统已安装了 Python 和 pip。您可以通过以下命令检查它们是否已安装:
python --version
pip --version
如果尚未安装,请使用以下命令安装 Python 和 pip:
对于 Ubuntu/Debian 系统:
sudo apt-get update
sudo apt-get install python3 python3-pip
对于 CentOS/RHEL 系统:
sudo yum install python3 python3-pip
- 接下来,根据您的系统架构选择合适的 PyTorch 版本。您可以在 PyTorch 官方网站(https://pytorch.org/get-started/locally/)上找到安装命令。以下是一些示例命令:
对于 Ubuntu/Debian 系统(CPU):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu111
对于 Ubuntu/Debian 系统(GPU,如果已安装 NVIDIA CUDA 工具包):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu111-gpu
对于 CentOS/RHEL 系统(CPU):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
对于 CentOS/RHEL 系统(GPU,如果已安装 NVIDIA CUDA 工具包):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu111-gpu
请将上述命令中的 cu111
替换为您系统上安装的 CUDA 版本。如果您不确定已安装的 CUDA 版本,请在终端中运行以下命令:
nvcc --version
- 安装完成后,您可以通过以下命令验证 PyTorch 是否已成功安装:
python3 -c "import torch; print(torch.__version__)"
如果安装成功,此命令将输出 PyTorch 的版本号。现在您可以在 Linux 系统中使用 PyTorch 了!