Skip to content

Clarify pip install instructions: GPU vs. CPU #10685

Description

Is your feature request related to a problem? Please describe.

I have seen quite a few issues posted where the poster has installed both pip install onnxruntime onnxruntime-gpu.

At least since September 2019 and #1826, as far as I understand it, if you have installed onnxruntime-gpu via pip there is no reason to install onnxruntime and you can switch to the CPU provider programmatically in Python, e.g. discussion in #486 and #7313.

Could it cause issues having both packages installed at once?

Describe the solution you'd like

https://onnxruntime.ai/docs/get-started/with-python.html would ideally be updated with an explicit comment stating that only onnxruntime-gpu should be installed to get CPU and GPU capabilities on machines with GPUs.

I believe at one point TensorFlow's documentation https://www.tensorflow.org/install/pip indicated something similar for tensorflow vs tensorflow-gpu meta-package, beforetensorflow-gpu became the same as tensorflow after 2.1.

For those who hadnt used tensorflow-gpu back then, the parallel distinction between onnxruntime vs. onnxruntime-gpu might be lost on them and hence it would be valuable to have explicit instructions in the ORT documentation.

Metadata

Metadata

Assignees

No one assigned

    Labels

    documentationimprovements or additions to documentation; typically submitted using template

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions