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/. ``` diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..bd3e656 --- /dev/null +++ b/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/.