22 lines
328 B
YAML
22 lines
328 B
YAML
name: cell_sim
|
|
channels:
|
|
- pytorch
|
|
- nvidia
|
|
- conda-forge
|
|
- defaults
|
|
dependencies:
|
|
- python=3.10
|
|
- xformers>=0.0.18
|
|
- torchmetrics
|
|
- torchdata
|
|
- omegaconf
|
|
- submitit
|
|
- fvcore
|
|
- iopath
|
|
- pip
|
|
- pip:
|
|
- --index-url https://download.pytorch.org/whl/cpu
|
|
- torch==2.3.0+cpu
|
|
- torchvision
|
|
|