Closed KorRyu3 closed 2 months ago
既存のrequirements.txtがCPUでしか動かないようになっていたので、GPU用に別のrequirementsを作成しました。
requirements.txt
onnxruntime
onnx-runtime-gpu
torch
torch+cu118
概要
既存の
requirements.txtがCPUでしか動かないようになっていたので、GPU用に別のrequirementsを作成しました。変更内容
全体
requirements-gpu.txtにて
onnxruntimeをonnx-runtime-gpuに変更torchをtorch+cu118に変更関連Issue, PR
その他