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
1 commit 3 branches 0 tags 383 KiB
Python 98.4%
Shell 1.6%
Find a file
Akemi Izuko 18426c7552
Init
2024-11-20 12:11:10 -07:00
cifar10-fast-simple Init 2024-11-20 12:11:10 -07:00
README.md Init 2024-11-20 12:11:10 -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.