安装 onnxruntime basicsr

本文介绍了安装ONNX Runtime的CPU和GPU版本(支持CUDA 11.x和12.x)的方法,以及安装basicsr 1.4.2的命令,包括使用国内镜像加速。

作者:zhuge···预计阅读 3 分钟·606 阅读·0 评论
安装 onnxruntime   basicsr

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

相关文章

评论

加载中...