Support for nvidia 3000 series GPUs
JamesDConley opened this issue · 3 comments
JamesDConley commented
Would love to run this, but had some trouble getting the GPU version running as it doesn't support my GPU. I tried using a newer version of the pytorch base image to get past this, but ran into some issues.
JamesDConley commented
I think I got this running for 3000 series cards after an hour and change of messing with the docker image. It's a little rough right now, but it's calculating features for the images super fast! If it runs all the way I'll clean it up a little and put in a PR if you'd like :)
johanmodin commented
That's great! Contributions are very welcome.
johanmodin commented
Fixed by #8. Thanks!