yolox module ‘torch._c‘ has no attribute ‘_cuda_setdevice‘

今天在使用yolox的时候训练报错了 module ‘torch._c’ has no attribute ‘_cuda_setdevice’，
请注意pytorch是否有安装对 是否安装的是gup支持的版本
如果是用的conda 下载的注意看下载的版本是否正确 ，我使用conda下载不好使 ，换成了pip下载就没报错了：pip install torch1.9.0+cu102 torchvision0.10.0+cu102 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html