如何在 AMD GPU 上使用 Flux.jl？
How to use Flux.jl on an AMD GPU?
我目前的笔记本电脑上没有 Nvidia GPU，但我仍然希望能够加快训练速度。 Per the Flux docs，它说开箱即用地支持 Nvidia GPU，但根本没有提到 AMD GPU。是否可以在 AMD GPU 上使用 Flux？
这在原则上应该是可行的，因为 AMDGPU.jl provides a similar interface for AMD GPUs as CUDA.jl does for NVIDIA GPUs, and Flux is claimed to be agnostic to array types. However, at best this will only work on Linux, since AMDGPU.jl relies on AMD's ROCm platform, and ROCm is only supported on Linux，因为 ROCm 确实是专门围绕 Linux 内核构建的。
我目前的笔记本电脑上没有 Nvidia GPU，但我仍然希望能够加快训练速度。 Per the Flux docs，它说开箱即用地支持 Nvidia GPU，但根本没有提到 AMD GPU。是否可以在 AMD GPU 上使用 Flux？
这在原则上应该是可行的，因为 AMDGPU.jl provides a similar interface for AMD GPUs as CUDA.jl does for NVIDIA GPUs, and Flux is claimed to be agnostic to array types. However, at best this will only work on Linux, since AMDGPU.jl relies on AMD's ROCm platform, and ROCm is only supported on Linux，因为 ROCm 确实是专门围绕 Linux 内核构建的。