🚑 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
|
||||
|
||||
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:
|
||||
|
|
Loading…
Reference in a new issue