Install ONNX Runtime CPU
pip install onnxruntime
Install ONNX Runtime GPU (CUDA 11.x)
The default CUDA version for ORT is 11.8.
pip install onnxruntime-gpu
Install ONNX Runtime GPU (CUDA 12.x)
For Cuda 12.x, please use the following instructions to install from ORT Azure Devops Feed
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
from https://onnxruntime.ai/docs/install/
安装 basicsr
pip install -i https://mirrors.aliyun.com/pypi/simple tb-nightly
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple basicsr==1.4.2