Unverified Commit ca560d86 authored by Martin Rünz's avatar Martin Rünz Committed by GitHub

Update INSTALL.md

parent 61ffdb38
...@@ -5,6 +5,7 @@ ...@@ -5,6 +5,7 @@
- torchvision from master - torchvision from master
- cocoapi - cocoapi
- yacs - yacs
- matplotlib
- (optional) OpenCV for the webcam demo - (optional) OpenCV for the webcam demo
...@@ -12,7 +13,7 @@ ...@@ -12,7 +13,7 @@
```bash ```bash
# maskrnn_benchmark and coco api dependencies # maskrnn_benchmark and coco api dependencies
pip install ninja yacs cython pip install ninja yacs cython matplotlib
# follow PyTorch installation in https://pytorch.org/get-started/locally/ # follow PyTorch installation in https://pytorch.org/get-started/locally/
# we give the instructions for CUDA 9.0 # we give the instructions for CUDA 9.0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment