PyTorch 是一个流行的深度学习框架,支持多种安装方式。以下是适用于 Linux 的安装步骤:


1. 使用 pip 安装

推荐通过官方预编译包安装,简单快捷:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

✅ 适用于 Python 3.8+ 环境
⚠️ 确保已安装 CUDA 驱动(如需 GPU 加速)

PyTorch_Pip_Installation

2. 使用 conda 安装

如果你使用 Anaconda 环境,可通过以下命令安装:

conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c conda-forge

📦 适合科学计算环境管理
🔗 了解更多 conda 安装细节

PyTorch_Conda_Installation

3. 从源码编译

需较高技术门槛,适合定制化需求:

git clone https://github.com/pytorch/pytorch.git  
cd pytorch  
git checkout v2.4.1  
python3 setup.py build --build-lib /path/to/install  

🛠️ 需安装 依赖库(如 CMake、Ninja 等)

PyTorch_Source_Code_Install

常见问题排查 🛠️


📌 提示:安装完成后,可通过以下命令验证:

python3 -c "import torch; print(torch.__version__)"

🎉 成功安装后,你将可以开始使用 PyTorch 进行机器学习实验!