diff --git a/Makefile b/Makefile index 732017e5..ebac34fd 100644 --- a/Makefile +++ b/Makefile @@ -48,7 +48,7 @@ cpplint-install: $(call check_pip_install,cpplint) clang-format-install: - command -v clang-format-11 || sudo apt-get install -y clang-format-11 + command -v clang-format || sudo apt-get install -y clang-format clang-tidy-install: command -v clang-tidy || sudo apt-get install -y clang-tidy @@ -85,7 +85,7 @@ cpplint: cpplint-install $(PYTHON) -m cpplint $(CXX_FILES) clang-format: clang-format-install - clang-format-11 --style=file -i $(CXX_FILES) -n --Werror + clang-format --style=file -i $(CXX_FILES) -n --Werror # Documentation @@ -111,7 +111,7 @@ lint: flake8 py-format mypy clang-format cpplint addlicense format: py-format-install clang-format-install addlicense-install $(PYTHON) -m isort --project torchopt $(PYTHON_FILES) $(PYTHON) -m yapf --in-place --recursive $(PYTHON_FILES) - clang-format-11 -style=file -i $(CXX_FILES) + clang-format -style=file -i $(CXX_FILES) addlicense -c $(COPYRIGHT) -l apache -y 2022 $(SOURCE_FOLDERS) clean-py: diff --git a/conda-recipe.yaml b/conda-recipe.yaml new file mode 100644 index 00000000..ba90aa56 --- /dev/null +++ b/conda-recipe.yaml @@ -0,0 +1,69 @@ +# Create virtual environment with command: +# +# conda env create --file conda-recipe.yaml +# + +name: torchopt + +channels: + - pytorch + - defaults + - nvidia + - conda-forge + +dependencies: + - python = 3.8 + + # Learning + - pytorch::pytorch = 1.11 + - pytorch::torchvision + - jax + - jaxlib + - tensorboard + - wandb + - pip: + - functorch + + # Device select + - nvidia::cudatoolkit = 11.3.1 + - cudnn + + # Build toolkit + - cmake >= 3.4 + - make + - cxx-compiler + - gxx >= 6.0, < 12.0 + - nvidia/label/cuda-11.3.1::cuda-minimal-build + - pybind11 + + # Misc + - pip + - typing-extensions + - numpy + - matplotlib-base + - seaborn + - python-graphviz + - pillow + + # Documentation + - sphinx + - sphinxcontrib-spelling + - sphinx-autobuild + - sphinx-copybutton + - sphinx_rtd_theme + + # Testing + - pytest + - pytest-cov + - pytest-xdist + - isort + - yapf + - mypy + - flake8 + - flake8-bugbear + - doc8 + - pydocstyle + - pyenchant + - clang-format + - clang-tools # clang-tidy + - cpplint