626_privacy/mia_on_model_distillation
2
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
Code for cmput 626 project fall 2024
13 commits 3 branches 0 tags 383 KiB
Python 98.4%
Shell 1.6%
Find a file
Akemi Izuko e4b5998dbb
Wres: dp training
2024-11-30 20:28:25 -07:00
cifar10-fast-simple Lira: remove deadcode 2024-11-23 23:32:39 -07:00
lira-pytorch Torchlira: add additional networks 2024-11-30 13:34:05 -07:00
wresnet-pytorch Wres: dp training 2024-11-30 20:28:25 -07:00
README.md Torchlira: cite repositories 2024-11-30 13:36:40 -07:00

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.