When USE_THRUST=ON
, unknown CUDA error happened:
File "/home/ubuntu/tvm/src/runtime/cuda/cuda_device_api.cc", line 108
CUDA: Check failed: e == cudaSuccess || e == cudaErrorCudartUnloading: unknown error
It can be reproduced with the following script
import numpy as np
import tvm
import xgboost
if __name__=='__main__':
a_np = np.ones((1, 16))
a_tvm = tvm.nd.array(a_np, ctx=tvm.gpu(0))
My environment: CUDA 10.0, thrust 1.9.3 XGBoost: 1.1.0 (installed with pip or from source)
I found a workaround is to import xgboost before importing TVM