ubuntu22.04安装rocm教程，AMD深度学习，6800xt配置pytorch、CUDA（已更新）本地安装torch

官方文档

对应torch下载地址

ROCm安装配置踩坑

• 遇到的问题
  • 安装ubuntu系统更新内核，内核新一点就可以
  • navi6800xt（gfx1030）显卡安装5.0及以上
  • rocm5.0以上是支持了navi显卡，如果是上一代的卡，可以装4.5这个版本系列，因为torch官网就有编译好的pytorch版本，可以直接安装到本地环境而不需要docker镜像
  • 安装后重启
  • （更新1）bios禁用安全启动。

ROCm安装

• （更新2）pytorch支持ROCM的版本更新到了5.4（pytorch2.0也支持），而rocm更新到了5.6（并且安装过程简化），似乎是支持大多数6000系gpu，笔记本的6000系列gpu（需要额外加一行参数）
  

• （更新3）ROCM现在已经支持windows，文章最开始的文档有官方的安装步骤，应该是挺简单的，各位有兴趣的可以试试

以下示例安装5.4，系统为ubuntu22.04，其他系统版本去官网看看安装说明找对应的安装脚本就可以

sudo apt update && sudo apt upgrade
sudo apt install wget gnupg2 
# 22.04安装这个
wget https://repo.radeon.com/amdgpu-install/5.4/ubuntu/jammy/amdgpu-install_5.4.50400-1_all.deb
sudo apt