Code for cmput 626 project fall 2024
cifar10-fast-simple | ||
lira-pytorch | ||
README.md |
Sources
Setup
Get miniconda here
conda create --name mia_distilled python=3.11.2
conda activate mia_distilled
conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
We've found that CUDA 12.2 will still run without issue on pytorch-cuda=12.1
.
There is also a pytorch-cuda=12.4
. Check your system CUDA version with
nvidia-smi
.