add environment.yml file

This commit is contained in:
Jack Langerman 2022-05-13 17:06:35 +00:00
parent ddfd4daa1b
commit b2a5f24d14

28
environment.yml Normal file
View file

@ -0,0 +1,28 @@
name: pointmlp
channels:
- pytorch
- nvidia
- conda-forge
dependencies:
# - cudatoolkit=10.2.89
- cudatoolkit=11.1
- cycler=0.10.0
- einops=0.3.0
- h5py=3.2.1
- matplotlib=3.4.2
- numpy=1.20.2
- numpy-base=1.20.2
- pytorch=1.8.1
- pyyaml=5.4.1
- scikit-learn=0.24.2
- scipy=1.6.3
- torchvision=0.9.1
- tqdm=4.61.1
- pip
- pip:
- pointnet2_ops_lib/.
# - pip:
# - torch >= 1.8
# - torchvision
# - numpy
# - kornia