mri-autoencoder-v0.1 / environment.yml
pidajay's picture
Added conda environment file for users to run notebook outside colab
b4b3a76
name: mri-autoencoder
channels:
- pytorch
- nvidia
- conda-forge
- defaults
dependencies:
- accelerate=0.25.0
- diffusers=0.24.0
- h5py=3.10.0
- ipykernel=6.26.0
- ipywidgets=8.1.1
- matplotlib=3.7.3
- nbconvert=7.12.0
- numpy=1.23.5
- omegaconf=2.3.0
- python=3.8
- pytorch::pytorch=1.13.1
- pytorch::torchvision=0.14.1
- scikit-image=0.19.3
- scipy=1.10.1
- tqdm=4.66.1
- transformers=4.36.0