• kaiJIN's avatar
    Support for running on arbitrary CUDA device. (#537) · f0318794
    kaiJIN authored
    * support for any one cuda device
    
    * Revert "support for any one cuda device"
    
    This reverts commit 0197e4e2ef18ec41cc155f3ae2a0face5b77e1e9.
    
    * support runnning for anyone cuda device
    
    * using safe CUDAGuard rather than intrinsic CUDASetDevice
    
    * supplement a header dependency (test passed)
    
    * Support for  arbitrary GPU device.
    
    * Support for arbitrary GPU device.
    
    * add docs for two method to control devices
    f0318794
Name
Last commit
Last update
.github/ISSUE_TEMPLATE Loading commit data...
configs Loading commit data...
demo Loading commit data...
docker Loading commit data...
maskrcnn_benchmark Loading commit data...
tests Loading commit data...
tools Loading commit data...
.flake8 Loading commit data...
.gitignore Loading commit data...
ABSTRACTIONS.md Loading commit data...
CODE_OF_CONDUCT.md Loading commit data...
CONTRIBUTING.md Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
MODEL_ZOO.md Loading commit data...
README.md Loading commit data...
TROUBLESHOOTING.md Loading commit data...
requirements.txt Loading commit data...
setup.py Loading commit data...