swh:1:snp:9112f69280e9abcc22402dd6eb4a1ab007a29028
Tip revision: ae88046b61c3d577bebd7cbab608cc914b50300b authored by Bastien DOIGNIES on 04 April 2024, 13:10:41 UTC
Updated requirements
Updated requirements
Tip revision: ae88046
environment.yml
name: qmcdiffusion
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
- python=3.11
- pytorch=2.1.1
- pytorch-cuda=12.1
- torchvision
- matplotlib
- scipy
- pot
- tensorboardx
- tqdm
- h5py
- pip
- pip:
- ema-pytorch