Closed VincentLeeMax closed 1 year ago
fix issue https://github.com/tkestack/gpu-manager/issues/169
when cuda >= 11.3, cuGetProcAddress could be the entrypoint, add init logic.
cuGetProcAddress
For example, running https://github.com/NVIDIA/cuda-samples/tree/v11.3/Samples/deviceQuery ,cudaGetDeviceCount will call cuGetProcAddress first.
cudaGetDeviceCount
fix issue https://github.com/tkestack/gpu-manager/issues/169
when cuda >= 11.3,
cuGetProcAddress
could be the entrypoint, add init logic.For example, running https://github.com/NVIDIA/cuda-samples/tree/v11.3/Samples/deviceQuery ,
cudaGetDeviceCount
will callcuGetProcAddress
first.