-
Python 3.11
-
Conda (Miniconda or Anaconda)
-
NVIDIA GPU (optional, for GPU acceleration)
-
CUDA-compatible drivers installed on your system
conda create -n automl python=3.11 cudatoolkit=11.3 -y
conda activate automl
Install the CUDA-enabled version of PyTorch:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
Note: Ensure the selected PyTorch/CUDA version is compatible with your GPU drivers.
pip install --upgrade pip setuptools wheel
pip install autogluon
pip install -r requirements.txt
Start the Streamlit application:
streamlit run app.py
To confirm that PyTorch can access your GPU:
python -c "import torch; print(torch.cuda.is_available())"
Expected output:
True
conda create -n automl python=3.11 cudatoolkit=11.3 -y
conda activate automl
pip install --upgrade pip setuptools wheel
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
pip install autogluon
pip install -r requirements.txt
streamlit run app.py