WSL How to guide - Use ROCm on Radeon GPUs

This guide walks you through the various installation processes required to pair ROCm™ with the latest high-end AMD Radeon™ 7000 series desktop GPUs, and get started on a fully-functional environment for AI and ML development.

How To

• Install Radeon software for WSL with ROCm
• Install PyTorch for Radeon GPUs
• Install ONNX Runtime for Radeon GPUs
• Install TensorFlow for Radeon GPUs
• Install Triton for Radeon GPUs

Note

• Use the Install Radeon software for WSL with ROCm section to get started

• Framework instructions (PyTorch, ONNX, TensorFlow, Triton) are the same as Linux

• MIgraphX and mGPU configuration are not currently supported by WSL