basicsr 安装问题

最新推荐文章于 2025-05-22 16:22:36 发布

会发paper的学渣

最新推荐文章于 2025-05-22 16:22:36 发布

阅读量3.1k

点赞数 9

CC 4.0 BY-SA版权

文章标签： python AIGC

本文链接：https://blog.csdn.net/sslfk/article/details/136190420

文章讲述了在安装basicsr时，由于包完整性问题可能导致安装失败，建议分开安装torch和torchvision，并指定CUDA版本。使用清华大学源和额外的PyTorch仓库来确保依赖包的完整性和兼容性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

basicsr依赖torch和torchvision和tb-nightly

安装时，如果使用单一源安装，由于原的包不完整性，容易导致安装失败：tb-nightly清华源不存在

所以建议安装方式为：

例如我的requirements内容如下：

basicsr==1.4.2
filetype==1.2.0
gradio==3.50.2
numpy==1.26.2
onnx==1.15.0
onnxruntime==1.16.3
opencv-python==4.9.0.80
psutil==5.9.6
realesrgan==0.3.0
torch==2.1.2
torchvision==0.16.2

其中torch根据自己的需要指定版本（不指定会使用最高版本）

如果Torch需要使用cuda版本，则需要单独安装：

安装方式如下(由于提前安装了torch，且requirements.txt中的torch版本一直，所以不会重新安装，在里面放入torch是为了约束basicsr依赖的torch版本，否则会安装最新版本)：

pip install torch=2.1.2 torchvision==0.16.2 -i https://download.pytorch.org/whl/cu118

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple --extra-index-url https://download.pytorch.org/whl/cu118