With the debut of ROCm 7.1.1, RDNA 4 users can now fully harness their GPUs for AI tasks on Windows, a significant step forward from previous Linux-only support.
Empowering AI on Windows
Previously, RDNA 4 users had to rely on Linux for native ROCm support or workarounds like ZLUDA or WSL. Now, full PyTorch support is accessible through ROCm on Windows, although TensorFlow, Jax, and ONNX remain Linux-exclusive for RDNA4. This advancement significantly boosts Radeon GPUs’ usability and performance for AI tasks like image generation, eliminating the cumbersome WSL requirement.
Initial Configuration
To get started, download and install the 25.20.0.17 drivers to enable 7.1.1 PyTorch support on Windows. While compatible with RDNA 3, RDNA 3.5, and RDNA 4, AMD currently lists only the 7900XTX from the RDNA 3 lineup as compatible. Next, install Miniconda, available here, and Git to clone the latest ComfyUI build. Create a new conda environment to manage dependencies efficiently. Use Python 3.12 for compatibility with ROCm 7.1.1 wheels for PyTorch.
Installing PyTorch for ROCm 7.1.1
To set up PyTorch, follow the detailed instructions provided here. Use the commands to install necessary ROCm packages and sub-dependencies. Once everything is installed, confirm by running a test script to verify GPU recognition. A command-line interface should acknowledge your GPU, like the 9060XT 8GB, confirming successful setup.
Installing ComfyUI
Choose a directory for ComfyUI installation, such as your Desktop, and clone the repository using Git. Navigate to the ComfyUI folder and run the command to install dependencies. Launch ComfyUI by executing python main.py, and access the GUI via the provided web server link.
Setting up models
In the GUI, locate the “SDXL Turbo” model under Templates and download it. Move the model into the ComfyUI > Models > Checkpoints directory. Add your creative prompt into the designated box and hit run to generate AI imagery locally on your RDNA 4 GPU. Engage with the community by sharing your creations in the comments!
Here’s an image created using the prompt: “beautiful landscape scenery glass bottle with a galaxy inside cute fennec fox snow HDR sunset.” Express your artistic flair and share your results!