19 lines
680 B
Markdown
19 lines
680 B
Markdown
## Sources
|
|
|
|
- [cifar10-fast-simple](https://github.com/99991/cifar10-fast-simple)
|
|
- [lira-pytorch](https://github.com/orientino/lira-pytorch)
|
|
- [wresnet-pytorch](https://github.com/AlexandrosFerles/Wide-Residual-Networks-PyTorch)
|
|
|
|
## Setup
|
|
|
|
Get miniconda [here](https://docs.anaconda.com/miniconda/install/#quick-command-line-install)
|
|
|
|
```bash
|
|
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`.
|