Create environment.yml

eigenvivek · Aug 8, 2024 · a9570ef · a9570ef
1 parent b24c0d3
commit a9570ef
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,8 @@
+name: diptorch
+channels:
+  - conda-forge
+  - pytorch
+  - nvidia
+dependencies:
+  - pip
+  - pytorch
diff --git a/settings.ini b/settings.ini
@@ -38,6 +38,8 @@ status = 3
 user = eigenvivek
 
 ### Optional ###
-requirements = torch
-dev_requirements = black
+requirements = 
+pip_requirements = torch
+conda_requirements = pytorch
+dev_requirements = nbdev black flake8 ipykernel ipywidgets jupyterlab jupyterlab_execute_time jupyterlab-code-formatter isort
 # console_scripts =