unisim icon indicating copy to clipboard operation
unisim copied to clipboard

WARNING:root:ONNX is running on CPU - you need to install CUDA/onnxruntime-gpu to run on GPU

Open fanghgit opened this issue 1 year ago • 1 comments

The code I run:

import os
import json
from copy import deepcopy
os.environ["BACKEND"] = "onnx"
import onnxruntime as ort
from unisim import TextSim

I got message: WARNING:root:ONNX is running on CPU - you need to install CUDA/onnxruntime-gpu to run on GPU.

However, ort.get_devices() gives "GPU".

How can I fix it? Many thanks in advance.

fanghgit avatar Dec 10 '24 14:12 fanghgit

Similar issue. Any feedback is appreciated.

FahadEbrahim avatar Mar 12 '25 17:03 FahadEbrahim