search

mlcommons / training

Reference implementations of MLPerf™ training benchmarks
https://mlcommons.org/en/groups/training
Apache License 2.0
1.6k stars 553 forks source link

[Mask R-CNN] Can this project work with CUDA 11 and Pytorch 1.9.0? #513

Closed ghfdjsk closed 1 year ago

ghfdjsk commented 3 years ago

Currently my machine has Cuda 11 and Pytorch 1.9.0. Due to the dependencies of other projects I am working on I can not change the version of these. However, I want to run the Mask R-CNN project. Is this project compatible with my system's version of Cuda and Pytorch? If so, then what changes should I make to run it? If not, how do I resolve this issue?

johntran-nv commented 1 year ago

In general our benchmarks may have specific library dependencies. This is partially to make sure we all get reproducible and fair results. Feel free to try running with other versions, and if this is critical to be compatible with specific other versions of the libraries, we can try to address it. In the future, I hope to containerize more of the references to help address some of these issues.

Feel free to reopen if this is still an important issue for you, but I'm closing for now.