YOLOX指定GPU训练

在train.py中的最上方,最上方添加以下代码:

import os
os.environ['CUDA_VISIBLE_DEVICES']='1'

YOLOX指定GPU训练_第1张图片

参考链接
https://blog.csdn.net/weixin_44570845/article/details/127108794

你可能感兴趣的:(YOLOX,python,开发语言)