From b2a5f24d14e78e0dce09b997bdcce2d90a157b8d Mon Sep 17 00:00:00 2001 From: Jack Langerman Date: Fri, 13 May 2022 17:06:35 +0000 Subject: [PATCH 1/3] add environment.yml file --- environment.yml | 28 ++++++++++++++++++++++++++++ 1 file changed, 28 insertions(+) create mode 100644 environment.yml diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..bb65779 --- /dev/null +++ b/environment.yml @@ -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 \ No newline at end of file From 53e06b31124e7d027fa4cc7516e9cd67e410517c Mon Sep 17 00:00:00 2001 From: Jack Langerman Date: Fri, 13 May 2022 18:48:11 +0000 Subject: [PATCH 2/3] remove unused code --- environment.yml | 5 ----- 1 file changed, 5 deletions(-) diff --git a/environment.yml b/environment.yml index bb65779..bd3e656 100644 --- a/environment.yml +++ b/environment.yml @@ -21,8 +21,3 @@ dependencies: - pip - pip: - pointnet2_ops_lib/. - # - pip: - # - torch >= 1.8 - # - torchvision - # - numpy - # - kornia \ No newline at end of file From d3b8f2326d6e47b46b746e9c48fa7686bc138c5e Mon Sep 17 00:00:00 2001 From: Jack Langerman Date: Fri, 13 May 2022 18:49:42 +0000 Subject: [PATCH 3/3] update readme --- README.md | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/README.md b/README.md index 5afd223..c424738 100644 --- a/README.md +++ b/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/. ```