pidajay commited on
Commit
b4b3a76
1 Parent(s): f47d81d

Added conda environment file for users to run notebook outside colab

Browse files
Files changed (1) hide show
  1. environment.yml +23 -0
environment.yml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: mri-autoencoder
2
+ channels:
3
+ - pytorch
4
+ - nvidia
5
+ - conda-forge
6
+ - defaults
7
+ dependencies:
8
+ - accelerate=0.25.0
9
+ - diffusers=0.24.0
10
+ - h5py=3.10.0
11
+ - ipykernel=6.26.0
12
+ - ipywidgets=8.1.1
13
+ - matplotlib=3.7.3
14
+ - nbconvert=7.12.0
15
+ - numpy=1.23.5
16
+ - omegaconf=2.3.0
17
+ - python=3.8
18
+ - pytorch::pytorch=1.13.1
19
+ - pytorch::torchvision=0.14.1
20
+ - scikit-image=0.19.3
21
+ - scipy=1.10.1
22
+ - tqdm=4.66.1
23
+ - transformers=4.36.0