add environment.yml file
This commit is contained in:
parent
ddfd4daa1b
commit
b2a5f24d14
28
environment.yml
Normal file
28
environment.yml
Normal 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
|
Loading…
Reference in a new issue