PyTorch Conda 安装指南

PyTorch 是一个开源的机器学习库，广泛用于深度学习领域。使用 Conda 安装 PyTorch 是一个简单且高效的方式。以下是基于 Linux 系统的安装步骤：

安装 Conda

首先，确保您的系统上已安装了 Anaconda 或 Miniconda。您可以从 Anaconda 官网或 Miniconda 官网下载并安装。

为了更好地管理您的项目，建议为 PyTorch 创建一个新的 Conda 环境。

conda create -n pytorch_env python=3.8

进入新创建的环境：

conda activate pytorch_env

然后，使用以下命令安装 PyTorch：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

请根据您的 CUDA 版本调整 cudatoolkit 的版本。

安装完成后，您可以通过以下命令验证 PyTorch 是否已成功安装：

import torch
print(torch.__version__)

如果输出 PyTorch 的版本号，则表示安装成功。