🚑 fix cuda deps, prevent future me from dying of unreproducibility

This commit is contained in:
Laurent FAINSIN 2023-05-30 11:45:36 +02:00
parent 241631628b
commit 377132fdd5

View file

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