hw1

We strongly suggest using conda (or miniconda) for package management. We tested this assignment using a conda environment with python 3.10.

conda create -n cmu16824hw1 python=3.10
conda activate cmu16824hw1
pip install -r requirements.txt

Note: If you are using a Blackwell GPU (e.g. RTX 5080 or 5090), you need a newer PyTorch version (2.7+) and CUDA 12.8+.