search

fanq15 / FSOD-code

MIT License
244 stars 61 forks source link

How can I run it when my cuda=10.2? #43

Open 1786939789 opened 4 years ago

1786939789 commented 4 years ago

My cuda is 10.2, so I can't install pytorch which is below 1.5.0. But it requires torch=0.4.1 and cuda=9.0 from readme.md. So is there any way to make me run this code?

fanq15 commented 4 years ago

You can try this code on COCO dataset https://github.com/fanq15/FewX.

vuducnghia commented 4 years ago

@1786939789 Does it work with cuda 10.2?