🚑 fix cuda deps, prevent future me from dying of unreproducibility
This commit is contained in:
parent
241631628b
commit
377132fdd5
|
@ -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:
|
||||||
|
|
Loading…
Reference in a new issue