Ubuntu22.04安装AMD显卡驱动和pytorch

一、安装pip和vim

sudo apt-get -y install python3-pip vim git

二、配置pip源

mkdir ~/.pip && vim ~/.pip/pip.conf

pip.conf

[global]
index-url = https://mirrors.cloud.tencent.com/pypi/simple
[install]
trusted-host = mirrors.cloud.tencent.com

三、安装显卡驱动和pytorch 

# rocm5.4.2需要Ubuntu内核5.15+
$ cd ~ && uname -srmv

• Ubuntu 22.04 + rocm 5.4.2

# 安装显卡驱动
$ wget https: