How To Install AMD ROCm, PyTorch, Stable Diffusion & YOLO on Linux

Описание

Today I’ll be doing a step by step guide showing how to install AMD’s ROCm on an RX 6000 series GPU, but this guide should also work for RX 5000 GPUs as well. I also show how to install PyTorch which is a framework for machine learning tools in Python such as Stable Diffusion and YOLO object detection which I’ll also be showing how to get set up today. If you’re an Nvidia user and are curious how to set up CUDA then I’ll be covering that in my next video.

Written Guide: https://phazertech.com/tutorials/rocm.html

Updated video guide (10/2023): https://youtu.be/BZMZ3351KDs

GPU used in this video:
XFX RX 6700 XT: https://amzn.to/3P1FE9i

These are affiliate links where I earn a small commission for purchases at no extra cost to you.
This is the easiest way to help the channel, thank you!
Amazon: https://amzn.to/484HUnU

Website: https://phazertech.com/

Donations
Buy me a coffee: https://www.buymeacoffee.com/phazertech
PayPal: https://www.paypal.com/donate/?business=HJJR9QWBLG67A&no_recurring=0&item_name=Thank+you+for+donating+to+Phazer+Tech%21&currency_code=USD

Chapters:
00:00 Intro
01:58 AMD Driver on Ubuntu
03:10 Other Requirements
05:48 PyTorch
06:43 YOLO Object Detection
08:00 Stable Diffusion Web UI
11:23 Drivers & PyTorch on Arch Distros
12:58 Outro