assert 'CUDAExecutionProvider' in onnxruntime.get_available_providers()
此时如果这里报错的时候,说明这里安装的onnxrumtime的对应版本不对
应该是
pip install onnxrumtime-gpu
而不是
pip install onnxruntime
assert 'CUDAExecutionProvider' in onnxruntime.get_available_providers()
此时如果这里报错的时候,说明这里安装的onnxrumtime的对应版本不对
应该是
pip install onnxrumtime-gpu
而不是
pip install onnxruntime