Code for cmput 626 project fall 2024
Find a file
2024-12-05 00:13:50 -07:00
cifar10-fast-simple Lira: remove deadcode 2024-11-23 23:32:39 -07:00
lira-pytorch fixed to use student model 2024-12-02 17:58:01 -07:00
one_run_audit O1: new data splitting 2024-12-05 00:13:50 -07:00
wresnet-pytorch Wres: clean up student a bit 2024-12-02 18:45:13 -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.