Skip to content

Enable Pytorch playbook on STX/KRK #279

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
danielholanda wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Enable Pytorch playbook on STX/KRK #279

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
fd88f91
Enable stx/krk on pytorch playbook
danielholanda May 20, 2026
ff175f0
Removing `amd-gpu-visible-linux` test
sreeram-11 May 22, 2026
264328b
Merge branch 'main' into dholanda/pytorch_krkstx
sreeram-11 May 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions playbooks/core/pytorch-rocm-llms/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,11 @@ SPDX-License-Identifier: MIT

## Overview

<!-- @github-only -->
> [!IMPORTANT]
> This playbook uses special tags that GitHub cannot render. Please visit [amd.com/playbooks](https://amd.com/playbooks) to correctly preview this content.
<!-- @github-only:end -->

Want to run powerful AI language models on your own hardware? This guide shows you how.
This tutorial uses PyTorch powered by AMD ROCm™ software to run models that can summarize documents, answer questions, generate text, and more, all running locally.

Expand Down
16 changes: 16 additions & 0 deletions playbooks/core/pytorch-rocm-llms/playbook.json
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,14 @@
"halo": [
"linux",
"windows"
],
"stx": [
"linux",
"windows"
],
"krk": [
"linux",
"windows"
]
},
"required_platforms": {
Expand All @@ -41,6 +49,14 @@
"halo": [
"windows",
"linux"
],
"stx": [
"linux",
"windows"
],
"krk": [
"linux",
"windows"
]
},
"difficulty": "beginner",
Expand Down
11 changes: 0 additions & 11 deletions playbooks/dependencies/driver.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,16 +29,5 @@ Download and install the latest AMD GPU driver for Linux:
1. Visit the [AMD Linux Drivers](https://www.amd.com/en/support/download/linux-drivers.html) page.
2. Follow the installation instructions provided on the download page.

<!-- @test:id=amd-gpu-visible-linux timeout=60 hidden=True -->
```bash
set -euo pipefail
sudo -n apt-get update -y
sudo -n apt-get install -y rocm-smi
rocm-smi
rocm-smi --showproductname
test -d /opt/rocm
test -e /opt/rocm/lib/libroctx64.so.4 -o -e /opt/rocm/lib/libroctx64.so
```
<!-- @test:end -->
<!-- @device:end -->
<!-- @os:end -->
Loading