ubuntu22.04 amd显卡 安装rocm教程

- 安装conda

conda download

下载Linux版本，进入服务器安装：

sh Anaconda3-2024.02-1-Linux-x86_64.sh

注意最后一步yes确认执行init

重新登录ssh后，在你的workspace创建python环境

conda create -n torch python=3.11
conda activate torch

执行python3确认版本号

这里可以看到是3.11，记住这个版本号

• 安装torch

torch download

执行上面复制的链接执行安装

进入python3命令行

python3
>>> import torch
>>> torch.__version__
'2.2.2+rocm5.7'

当前 rocm 版本为 5.7, torch 版本为 2.2.2，记住当前版本号

• 安装本地whl

whl download
筛选 torch 版本为上面安装的版本： torch2.2.2 对应 torch-2.2.2
筛选 rocm 版本为上面安装的版本： rocm5.7 对应 rocm5.7
筛选 python 版本为上面安装的版本： python3.11 对应 cp311

下载后安装

pip3 install torch-2.2.2+rocm5.7-cp311-cp311-linux_x86_64.whl

如果出错一般是下载的whl 的 文件跟 python 和 rocm torch 版本对不上

确认成功

Python 3.11.7 (main, Dec 15 2023, 18:12:31) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
True

返回 true 即可
非root用户为false时，执行

sudo usermod -a -G render $LOGNAME

关于显卡驱动

参考 https://blog.csdn.net/qq_51403540/article/details/123951460