Skip to content

Latest commit

 

History

History
13 lines (9 loc) · 611 Bytes

dependencies.md

File metadata and controls

13 lines (9 loc) · 611 Bytes

It's mainly the same as: https://github.com/eriklindernoren/PyTorch-YOLOv3

conda create -n test_yolo python=3.7
conda activate test_yolo
pip install -r requirements.txt

Finally, install torch>=1.0 & torchvision.
Please go to "https://pytorch.org/" and install the one that matches with your computer.
I'm using Stable(1,1), Linux, Pip, Python 3.7, CUDA 10.0, and torch-1.1.0, torchvision-0.3.0

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-linux_x86_64.whl
pip install https://download.pytorch.org/whl/cu100/torchvision-0.3.0-cp37-cp37m-linux_x86_64.whl