pytorch 的 dataset 中使用 onnxruntime

如果在 dataset 中预处理图像时，用到了 onnxruntime 的 cudaep，出现这样的错误：

1、2022-12-13 13:53:01.554864883 [E:onnxruntime:Default, cuda_call.cc:119 CudaCall] CUDA failure 3: initialization error ; GPU=32765 ; hostname=ubuntu16 ; expr=cudaSetDevice(GetDeviceId()); 
2、onnxruntime.capi.onnxruntime_pybind11_state.Fail: Caught Fail in DataLoader worker process 0.
3、onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : CUDA error executing cudaSetDevice(GetDeviceId())
4、TypeError: cannot pickle 'onnxruntime.capi.onnxruntime_pybind11_state.InferenceSession' object

...

把 DataLoader 中 num_workers 设置成 0 就好了。