File size: 582 Bytes
738cc8c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | name: FoMo4Wheat
channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
- defaults
- pytorch
- nvidia
- xformers
- conda-forge
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
dependencies:
- python=3.11
- pytorch::pytorch=2.5.1
- pytorch::pytorch-cuda=12.1
- pytorch::torchvision
- omegaconf
- torchmetrics
- fvcore
- iopath
- xformers::xformers
- pip
- pip:
- git+https://github.com/facebookincubator/submitit
- --extra-index-url https://pypi.nvidia.com
- cuml-cu11
|