PointMLP/environment.yml
2023-08-03 16:38:02 +02:00

34 lines
407 B
YAML

name: pointmlp
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
#---# basic python
- pytorch
- tqdm
- numpy
- scipy
- scikit-learn
#---# file readers
- h5py
- pyyaml
#---# tooling (linting, typing...)
- ruff
- mypy
- black
- isort
#---# visu
- matplotlib
#---# pytorch
- cudatoolkit
- cycler
- einops
- torchvision
- pip
- pip:
- pointnet2_ops_lib/.