Code for cmput 626 project fall 2024
Find a file
2024-12-04 07:45:06 -07:00
cifar10-fast-simple Lira: remove deadcode 2024-11-23 23:32:39 -07:00
lira-pytorch made to use student/teacher subsets that are half data each 2024-12-04 07:25:56 -07:00
one_run_audit O1: add epsilon audit 2024-12-03 23:26:50 -07:00
wresnet-pytorch added if statement to either keep the datapoint in our out 2024-12-04 07:45:06 -07:00
env.yaml Wres: add conda env file 2024-12-02 19:01:26 -07:00
README.md Torchlira: cite repositories 2024-11-30 13:36:40 -07:00

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.