Skip to content

Commit

Permalink
add environment.yml file
Browse files Browse the repository at this point in the history
  • Loading branch information
JackLangerman committed May 13, 2022
1 parent ddfd4da commit b2a5f24
Showing 1 changed file with 28 additions and 0 deletions.
28 changes: 28 additions & 0 deletions environment.yml
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit b2a5f24

Please sign in to comment.