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 =