From a9570efb849c67b170b94781e485fb45082c2a9e Mon Sep 17 00:00:00 2001 From: Vivek Gopalakrishnan Date: Thu, 8 Aug 2024 07:10:41 -0400 Subject: [PATCH] Create environment.yml --- environment.yml | 8 ++++++++ settings.ini | 6 ++++-- 2 files changed, 12 insertions(+), 2 deletions(-) create mode 100644 environment.yml diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..02963e5 --- /dev/null +++ b/environment.yml @@ -0,0 +1,8 @@ +name: diptorch +channels: + - conda-forge + - pytorch + - nvidia +dependencies: + - pip + - pytorch \ No newline at end of file diff --git a/settings.ini b/settings.ini index 2132910..bbaab8e 100644 --- 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 =