Traceback (most recent call last):


 "train.py", line 633, in <module>

torch.cuda.set_device(opt.local_rank)


 "/opt/conda/lib/python3.7/site-packages/torch/cuda/__init__.py", line 281, in set_device

torch._C._cuda_setDevice(device)
RuntimeError: cuda runtime error (101) : invalid device ordinal at /opt/conda/conda-bld/pytorch_1595629403081/work/torch/csrc/cuda/Module.cpp:59

Traceback (most recent call last):


 ".../test.py", line 48, in <module>

res = model.predict({'myinput__0': input_array})


 ".../lib/python3.6/site-packages/coremltools/models/model.py", line 326, in predict

return self.__proxy__.predict(data,useCPUOnly)
RuntimeError: {
