Open zrr1991 opened 4 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
非常感谢
大神请问您使用的版本都是啥啊: 我看到您使用了docker, cuda10.2, torch1.5 请问您ubuntu的版本,可以给搭建环境教程,和您的YOLOV4的使用教程吗