注意

如果您在 Windows 上安装 AutoGluon 时遇到困难，请在此 GitHub Issue 中提供详细信息。

在 Windows 上安装 AutoGluon 建议使用 Anaconda

  1. 安装 Anaconda

    • 如果已安装 Anaconda 且版本较旧，请按照本指南进行更新

  2. 打开 Anaconda Prompt (anaconda3)

  3. 在 Anaconda Prompt 中，执行以下操作

conda create -n myenv python=3.11 cudatoolkit=11.3 -y
conda activate myenv

  1. 按照 PyTorch 安装文档 安装正确的 GPU PyTorch 版本（推荐）。或者，使用以下命令

pip install torchvision==0.19.1 --force-reinstall --extra-index-url https://download.pytorch.org/whl/cu121

  1. 通过在 Python 中测试来健全性检查您的安装是否有效并能检测到您的 GPU

import torch
print(torch.cuda.is_available())  # Should be True
print(torch.cuda.device_count())  # Should be > 0

  1. 继续使用上面创建的 conda 环境执行剩余的安装步骤