pytorch cuda等版本 - Githubissues

argusswift / YOLOv4-pytorch

This is a pytorch repository of YOLOv4, attentive YOLOv4 and mobilenet YOLOv4 with PASCAL VOC and COCO

1.68k stars 329 forks source link

Open goldentimecoolk opened 3 years ago

goldentimecoolk commented 3 years ago

您好，请问您使用的pytorch版本是多少呀？

Leer-rui commented 2 years ago

刚开始我按照IMSTALL.md的要求安装cudatoolkit=10.0，不过运行train.py发现需要将pytorch升级为1.6.0，pytorch=1.6和cudatoolkit=10.0并不兼容，因此可能需要将cudatoolkit换为10.1版本或更高。