- Python 3.9+ & pip
- Git
# Create venv
python -m venv venv
# macOS/Linux:
source venv/bin/activate
# Windows:
venv\Scripts\activate
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128# Installs everything, including CPU PyTorch (or GPU if detected & compatible)
pip install -r requirements.txt- (Local GPU Only): If you need a specific CUDA version locally, install PyTorch first from pytorch.org, then run
pip install -r requirements.txt.
# Create a test.py file with this code to verify PyTorch GPU setup
import torch
print(f"Is CUDA available: {torch.cuda.is_available()}")
print(f"Current device: {torch.cuda.get_device_name(0)}" if torch.cuda.is_available() else "No CUDA device available")
streamlit run main.py Notes:
- Deactivate environment:
deactivate(venv) orconda deactivate(conda).