Skip to content
This repository was archived by the owner on May 10, 2026. It is now read-only.
This repository was archived by the owner on May 10, 2026. It is now read-only.

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

@fanghgit

Description

@fanghgit

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions