assert 'CUDAExecutionProvider' in onnxruntime.get_available_providers()

此时如果这里报错的时候,说明这里安装的onnxrumtime的对应版本不对
应该是

pip install onnxrumtime-gpu

而不是

pip install onnxruntime

更多推荐

assert ‘CUDAExecutionProvider‘ in onnxruntime.get_available_providers()