diff --git a/environment.yml b/environment.yml index 4b0211b..0911f67 100644 --- a/environment.yml +++ b/environment.yml @@ -1,28 +1,37 @@ name: sap + channels: - - pytorch3d + - nodefaults - conda-forge - pytorch - - defaults + - nvidia + - pyg + dependencies: - - python - - pytorch - - torchvision - - cudatoolkit - - numpy - - matplotlib - - pyyaml - - scipy +#---# basic python + - python=3.8 - tqdm + - pyyaml +#---# visu + - matplotlib +#---# scientific + - numpy + - scipy - trimesh - igl - - tensorboard - - pytorch3d +#---# pytorch + - pytorch + - pytorch-cuda=11.8 + - cudatoolkit + - torchvision + - torch-scatter #---# tooling (linting, typing...) - ruff - mypy - black - isort +#---# logging + - tensorboard #---# pip shit - pip - pip: