How to initialize onnx input CreateTensor with gpu meory instead of CreateCpu #19240
Labels
ep:CUDA
issues related to the CUDA execution provider
platform:windows
issues related to the Windows platform
stale
issues that have not been addressed in a while; categorized by a bot
Describe the issue
hi,How to initialize onnx input CreateTensor with gpu meory instead of CreateCpu?I haven't found a solution yet。
To reproduce
Ort::MemoryInfo memory_info = Ort::MemoryInfo::CreateCpu(OrtArenaAllocator, OrtMemTypeDefault);
//Ort::MemoryInfo memory_info = Ort::MemoryInfo::CreateCuda(OrtDeviceAllocator, 0);
Ort::Value input_tensor = Ort::Value::CreateTensor(memory_info, input_tensor_values.data(),
input_tensor_size, input_node_dims.data(), 4);
Urgency
No response
Platform
Windows
OS Version
win10
ONNX Runtime Installation
Released Package
ONNX Runtime Version or Commit ID
1.8.1 or 1.16
ONNX Runtime API
C++
Architecture
X64
Execution Provider
CUDA
Execution Provider Library Version
No response
The text was updated successfully, but these errors were encountered: