julia - 如何在 AMD GPU 上使用 Flux.jl？

Question

我目前的笔记本电脑上没有 Nvidia GPU，但我仍然希望能够加快训练速度。根据 Flux docs，它说 Nvidia GPU 是开箱即用的，但根本没有提到 AMD GPU。是否可以在 AMD GPU 上使用 Flux？

Answer 1

原则上这应该是可能的，因为AMDGPU.jl为 AMD GPU 提供了与CUDA.jl为 NVIDIA GPU提供的类似接口，并且声称 Flux与数组类型无关。但是，这充其量只能在 Linux 上运行，因为 AMDGPU.jl 依赖于 AMD 的 ROCm 平台，而ROCm 仅在 Linux 上受支持，因为 ROCm 确实是专门围绕 Linux 内核构建的。