Please use Python 3.8.
Run the following script to setup all dependencies for the assignment:
chmod +x setup.sh
./setup.sh /path/to/python_env/lib/python3.8/site-packages # e.g. ./download_cub.sh ~/miniconda3/envs/vlr_hw2/lib/python3.8/site-packages