Add conda env

sarahkpardo · Jul 21, 2021 · 09a6503 · 09a6503
1 parent 7ccca4b
commit 09a6503
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -7,7 +7,5 @@ Neural networks designed for operator-valued functional learning problems.
 A conda environment with required modules can be installed using
 
 ```
-conda create -n neural_operators --file operators-env.txt
+conda env create -f operators-env.yml
 ```
-
-# Usage
diff --git a/operators-env.yml b/operators-env.yml
@@ -0,0 +1,11 @@
+name: operators-env
+channels:
+  - defaults
+dependencies:
+  - python=3.8
+  - torchaudio==0.7.0
+  - pytorch==1.7.0
+  - torchvision==0.8.0
+  - cudatoolkit=10.2
+  - einops
+  - networkx