626_privacy/mia_on_model_distillation
2
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions
mia_on_model_distillation/README.md

680 B
Raw Blame History

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.