utils/torch_utils.py · 8ace1b1b992433f31721c0553287cd664f2efe6b · Administrator / yolov5 · GitLab

Blame History Permalink

PyTorch Hub models default to CUDA:0 if available (#2472) · 9b11f0c5
由 Glenn Jocher 提交于 3月 15, 2021
```
* PyTorch Hub models default to CUDA:0 if available

* device as string bug fix
```
9b11f0c5

torch_utils.py 11.8 KB