Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ma-xu committed Apr 25, 2022
1 parent 07195fe commit 6ce676e
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -74,8 +74,9 @@ cd pointMLP-pytorch
conda create -n pointmlp python=3.7 -y
conda activate pointmlp

# 3. install required libs, pytorch 1.8.1, torchvision 0.9.1, etc.
pip install -r requirements.txt
# 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

# 4. install CUDA kernels
pip install pointnet2_ops_lib/.
Expand Down

0 comments on commit 6ce676e

Please sign in to comment.