Added conda environment file for users to run notebook outside colab
Browse files- 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
|