Skip to content

Instantly share code, notes, and snippets.

View pochenyun's full-sized avatar

pochenyun pochenyun

2 followers · 9 following
View GitHub Profile
@jurgonaut
jurgonaut / amd-rocm-torch-ubuntu.md
Last active January 10, 2025 08:35
Install AMD GPU ROCm and PyTorch on Ubuntu

This is a short guide on how to setup ROCm on Ubuntu. The most important step is that when we install PyTorch we get the latest available ROCm version for it.

Prerequisites

  • Python 3.10 or newer
  • Ubuntu
  • Make sure you have the appropriate kernel version, see prerequesites.

Amdgpu

Select the desired amdgpu version and download the correct Ubuntu version, replace <version> and <codename>. Check the available versions first https://repo.radeon.com/amdgpu-install/.