fix windows cuda support for python 3.8 + #6046
Conversation
|
Had a discussion with George, he said:
Therefore, to make the pipelines fully work, we need to update the CI build machines to
|
for consistency, i don't think we need to deprecate cudnn_home just for windows. it can point to same location as cuda_home |
I don't think this is true. #5956 did work fine for onnxruntime-gpu when CUDA/cuDNN are properly installed. The problem with that fix was that it didn't work for CPU version of onnxruntime package (complaining about missing CUDA/cuDNN), which wasn't caught because the relevant tests were disabled at the time. The second statement should be
Therefore, there was no reason to move os.add_dll_directory() to from |
yes. it is true. #5956 did not work for a python cuda build with python 3.8+ on Windows. and if you look at capi._pybind_state.py , and that is where you will get the error i pasted above import onnxruntime as ort |
#5956 doesn't work.
os.add_dll_directory() must be called before onnxruntime.capi._pybind_state gets imported.
otherwise you will get