18 lines
530 B
Markdown
18 lines
530 B
Markdown
|
## Sources
|
||
|
|
|
||
|
|
- [cifar10-fast-simple](https://github.com/99991/cifar10-fast-simple)
|
||
|
|
|
||
|
|
## 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`.
|