conda.yaml

name: mit-py311
channels:
- conda-forge
- pytorch
- nvidia
dependencies:
- python==3.11
- pytorch==2.2.2
- torchvision==0.17.2
- torchaudio==2.2.2
- pytorch-cuda=12.1
- numpy<2