environment

krrish94 · Apr 16, 2020 · 5478816 · 5478816
1 parent 90ddaf2
commit 5478816
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,21 @@
+name: nerf
+channels:
+  - pytorch
+  - conda-forge
+  - defaults
+dependencies:
+  - python>=3.7
+  - pip
+  - pytorch>=1.3
+  - cuda100
+  - cudatoolkit>=10.1
+  - cudatoolkit-dev
+  - imageio
+  - numpy
+  - pyyaml
+  - tensorboard
+  - torchvision
+  - tqdm
+  - pip:
+    - opencv-python-headless
+    - "git+https://github.com/aliutkus/torchsearchsorted#egg=torchsearchsorted"