如何在 ubuntu 22.04 中安装合适的 Nvidia 驱动程序在具有 AMD radeon GPU 的设备上运行 YOLOR

2024-6-7 • tag-icon

如何在 ubuntu 22.04 中安装合适的 Nvidia 驱动程序在具有 AMD radeon GPU 的设备上运行 YOLOR

现在我正在一台配备 AMD Radeon GPU 的笔记本电脑上使用双系统 ubuntu 22.04，并尝试安装 Nvidia 驱动程序来运行约洛尔。在步骤中，我需要安装推荐的Nvidia 驱动程序。
在说明中，我应该使用行找到推荐的 Nvidia 驱动程序
ubuntu-drivers devices，但执行该行没有显示任何内容。
我怀疑发生这种情况是因为我没有 Nvidia 显卡。那么，我如何才能为我的带有 AMD Radeon GPU 的笔记本电脑找到并安装合适且推荐的 Nvidia 驱动程序呢？

我知道我可能无法在我的 AMD radeon GPU 上安装 Nvidia 驱动程序。我只是想找到一种方法让 YOLOR 工作。但似乎找不到如何在没有安装 Nvidia 驱动程序的情况下进行设置的说明。

输出sudo lshw -c video：

*-display                 
       description: VGA compatible controller
       product: Lucienne
       vendor: Advanced Micro Devices, Inc. [AMD/ATI]
       physical id: 0
       bus info: pci@0000:05:00.0
       logical name: /dev/fb0
       version: c1
       width: 64 bits
       clock: 33MHz
       capabilities: pm pciexpress msi msix vga_controller bus_master cap_list fb
       configuration: depth=32 driver=amdgpu latency=0 resolution=1920,1080
       resources: irq:57 memory:d0000000-dfffffff memory:e0000000-e01fffff ioport:f000(size=256) memory:fcc00000-fcc7ffff

安装说明：

Ubuntu：下载 Yolor Github https://github.com/WongKinYiu/yolor 安装步骤：

首先安装 Python

sudo apt 更新
sudo apt install -y zip htop屏幕 libgl1-mesa-glx
sudo apt-get 安装 python-is-python3
pip 安装 seaboen thop

下载 Cuda 深度学习模型套件

https://developer.nvidia.com/cuda-downloads
Linux -> x86_64 -> Ubuntu -> 22.04 -> deb (本地)
按照说明下载

下载 Nvidia 驱动程序

https://zh-tw.linuxcapable.com/how-to-install-and-upgrade-nvidia-drivers-on-ubuntu-20-04/ 最后使用nvidia-smi命令查看Cuda是否安装

cd 到 yolor 的文件夹

pip install -r requirement.txt （应该会遇到torch版本问题）

cd 到根路径（出现 ~ ）

pip 安装 torch
pip 安装 torchvision
最后，使用 pip list 查看有哪些可用的包，
然后用Requirement.txt检查包/版本是否安装

下载 yolor_p6.pt
测试模型

python detect.py --source inference/images/horses.jpg --cfg cfg/yolor_p6.cfg --weights yolor_p6.pt --conf 0.25 --img-size 1280 --device 0 （Mish_cuda/pytorch_wavelets 无需安装）

任何帮助都值得感激，谢谢！

相关内容