codefire53 / pytorch-intro

PyTorch Intro

Environment Installation

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

Dataset:

Download the dataset from Kaggle using the guide under dataset/download_dataset.txt