[BUG] ...COLAB GPU not detected in ONNX model

[sinbrolly]

Couldn't run onnx-gpu u2net on COLAB gpu.

!pip install rembg[gpu] -qU
!pip install onnxruntime-gpu==1.12.0 -qU

import torch
torch.cuda.is_available()
True

import onnxruntime as ort
ort.get_available_providers()
[
'TensorrtExecutionProvider',
 'CUDAExecutionProvider', 
'CPUExecutionProvider'
]

from rembg import remove, new_session
session = new_session('u2net')
Downloading data from 'https://github.com/danielgatis/rembg/releases/download/v0.0.0/u2net.onnx' to file '/root/.u2net/u2net.onnx'.
100%|████████████████████████████████████████| 176M/176M [00:00<00:00, 102GB/s]

session.inner_session.get_providers()
['CPUExecutionProvider']

[github-actions[bot]]

This issue is stale because it has been open for 30 days with no activity.

[github-actions[bot]]

This issue was closed because it has been inactive for 14 days since being marked as stale.