Run the following to create the environment:
conda create --name pytorch-intro python=3.9
conda activate pytorch-intro
conda install pytorch==1.12.1 torchvision==0.13.1 cudatoolkit=11.3 -c pytorch
pip install tqdm
Download the dataset from Kaggle using the guide under dataset/download_dataset.txt