无法在 Ubuntu(WSL2) 中访问 AMD GPU(7900xtx)
作者: Dr-yangyangzhang创建于 2024年12月16日更新于 2024年12月16日
现在 AMD 正式支持 WSL2 中的 GPU。我按照官方说明在 WSL2 系统(Ubuntu 22.4 LTS)中安装了驱动。之后,我创建了一个虚拟环境来安装 nougat-orc。由于说明中特别指出,只有 Windows 用户在安装 nougat-orc 之前才应安装 pytorch,因此我没有手动安装 pytorch。之后,我可以正常运行 nougat,但它无法检测到 GPU。我想知道我是否需要执行其他步骤。
内容来源: facebookresearch/nougat