WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.88k stars 585 forks source link

环境版本 #146

Open zrr1991 opened 4 years ago

zrr1991 commented 4 years ago

大神请问您使用的版本都是啥啊: 我看到您使用了docker, cuda10.2, torch1.5 请问您ubuntu的版本,可以给搭建环境教程,和您的YOLOV4的使用教程吗

WongKinYiu commented 3 years ago
# create the docker container, you can change the share memory size if you have more.
nvidia-docker run --name yolov4_csp -it -v your_coco_path/:/coco/ -v your_code_path/:/yolo --shm-size=64g nvcr.io/nvidia/pytorch:20.06-py3

# install mish-cuda, if you use different pytorch version, you could try https://github.com/JunnYu/mish-cuda
cd /
git clone https://github.com/thomasbrandon/mish-cuda
cd mish-cuda
python setup.py build install

# go to code folder
cd /yolo
zrr1991 commented 3 years ago

非常感谢