Merge pull request #31 from JackLangerman/evn_file

add environment.yml file
This commit is contained in:
Xu Ma 2022-05-14 21:24:44 -04:00 committed by GitHub
commit 9ddef814b0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 24 additions and 8 deletions

View file

@ -71,15 +71,8 @@ git clone https://github.com/ma-xu/pointMLP-pytorch.git
cd pointMLP-pytorch cd pointMLP-pytorch
# 2. create a conda virtual environment and activate it # 2. create a conda virtual environment and activate it
conda create -n pointmlp python=3.7 -y conda env create
conda activate pointmlp conda activate pointmlp
# 3. install required libs, pytorch1.7+, torchvision, etc.
conda install pytorch==1.10.1 torchvision==0.11.2 cudatoolkit=10.2 -c pytorch -y
pip install cycler einops h5py pyyaml==5.4.1 scikit-learn==0.24.2 scipy tqdm matplotlib==3.4.2
# 4. install CUDA kernels
pip install pointnet2_ops_lib/.
``` ```

23
environment.yml Normal file
View file

@ -0,0 +1,23 @@
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/.