Merge pull request #31 from JackLangerman/evn_file

add environment.yml file

README.md

@@ -71,15 +71,8 @@ git clone https://github.com/ma-xu/pointMLP-pytorch.git
 cd pointMLP-pytorch
 
 # 2. create a conda virtual environment and activate it
-conda create -n pointmlp python=3.7 -y
+conda env create
 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/.
 ```
 
 

environment.yml

@@ -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/.