在安装PyTorch时避免安装NVIDIA相关包的指南

阅读提示：本文共计约1450个文字，预计阅读时间需要大约4分钟，由作者免费领养编辑整理创作于2023年11月05日23时31分52秒。

内容：

在安装 PyTorch 时，有时我们可能不需要安装与 NVIDIA GPU 加速相关的包。以下是避免安装 NVIDIA 相关包的步骤：

1. 确保已安装 Python 和 CUDA 版本。根据您选择的 PyTorch 版本，您需要安装相应的 CUDA 版本。例如，如果您选择安装 PyTorch 1.8，那么您需要安装 CUDA 11.0。

2. 从 PyTorch 官方网站下载对应的安装包。访问 https://pytorch.org/get-started/locally/ 并选择与您系统兼容的安装包。

3. 使用命令行或终端打开安装包所在的文件夹。

4. 运行以下命令（将 <version> 替换为您的 PyTorch 版本）：

pip install torch==<version> cu<CUDA_VERSION> -f https://download.pytorch.org/whl/lts/1.8/

这里，cu<CUDA_VERSION> 表示您需要的 CUDA 版本。例如，如果您的 CUDA 版本是 11.0，则应输入 cu110。

5. 如果您没有 NVIDIA GPU，或者不需要 GPU 加速，可以跳过此步骤。如果需要 GPU 加速，请确保安装了正确的驱动程序。

6. 在安装完成后，您可以使用以下命令检查是否成功安装了 PyTorch：

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

如果输出显示与所安装版本相符的 PyTorch 版本，则表示已成功安装。