Ubuntu+AMD显卡搭建Real-ESRGAN

realJayce

已于 2023-03-29 07:09:50 修改

阅读量493

点赞数

文章标签： ubuntu linux 运维

于 2023-03-28 15:04:25 首次发布

本文链接：https://blog.youkuaiyun.com/u013636118/article/details/129815508

版权

本文介绍了如何在Ubuntu20.04系统上，利用ROCM5.2和root权限搭建Real-ESRGAN，包括安装PyTorch和其他依赖库，如BasicsR、FaceXLib和GFPGAN。该过程适用于AMD显卡，并提供了命令行步骤来执行图像和视频的修复与增强操作。

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

目前只在Ubuntu20.04+rocm5.2上搭建成功，需要切换到root权限搭建，否则Real-ESRGAN会自动下载nvidia的驱动（可能需要添加到环境变量中），搭建链接：Ubuntu22.04安装AMD显卡驱动和pytorch_V2077的博客-优快云博客，项目链接：GitHub - xinntao/Real-ESRGAN: Real-ESRGAN aims at developing Practical Algorithms for General Image/Video Restoration.

安装Real-ESRGAN

$ cd ~
$ git clone https://github.com/xinntao/Real-ESRGAN.git
$ cd Real-ESRGAN

# 安装 basicsr - https://github.com/xinntao/BasicSR
# 我们使用BasicSR来训练以及推断
$ pip install basicsr
# facexlib和gfpgan是用来增强人脸的
$ pip install facexlib
$ pip install gfpgan
$ pip install -r requirements.txt
$ sudo python3 setup.py develop<

最低0.47元/天解锁文章