ort icon indicating copy to clipboard operation
ort copied to clipboard

torch-ort cannot be installed on windows: onnxruntime-training not found

Open alpatots opened this issue 2 years ago • 5 comments

When running pip install torch-ort in a conda environment on Windows, I get the following error:

ERROR: Could not find a version that satisfies the requirement onnxruntime-training (from versions: none) ERROR: No matching distribution found for onnxruntime-training

However, if I run the same command in a conda environment in WSL, it works just fine. Other people could repro. Seems to be a Windows issue.

alpatots avatar Jun 22 '22 21:06 alpatots

Yes, as of now, onnxruntime-training (ORTModule) is not available on Windows. We do plan to change this in the near future. But I don't have a timeline to share with you yet.

baijumeswani avatar Jul 18 '22 21:07 baijumeswani

@alpatots : Can you add some details for your scenario? Are you looking for Windows packages with Nvidia GPU support?

askhade avatar Jul 28 '22 23:07 askhade

@askhade I don't think I have more details than that... Our package takes a dependency on torch-ort, and we discovered it made our package un-installable on Windows. It does not affect runtime because we run jobs on linux anyways.

alpatots avatar Jul 28 '22 23:07 alpatots

@askhade @baijumeswani There are many machine vision application based on Windows, currently the machine vision application will add Pytorch model (eg, Mask RCNN)for defects detection. We retain the Pytorch model on Windows and want to deploy it by torch-ort on Intel ARC GPU on Windows as well. We meet the same problem when install the torch-ort-infer[openvino] on Windows No module named 'onnxruntime.training'

Looking forward to add Windows support ASAP! Thank you very much in advance!!

dlod-openvino avatar Nov 23 '22 07:11 dlod-openvino

Add error screenshot when execute:

from torch_ort import ORTInferenceModule, OpenVINOProviderOptions

No module named 'onnxruntime.training'

dlod-openvino avatar Nov 23 '22 07:11 dlod-openvino