目前在Linux系统上,消费者级显卡中,AMD官方新版ROCm只支持RX7900XTX、RX7900GRE等几款性能强的。稍微弱一点的消费者级显卡早已不再官方支持。RX580这种老显卡就更不提供官方支持了。
好在AMD的许多软件代码是开源的,官方不支持,开源世界的人们也可以想办法搞定。所以,今天给大家带来RX580显卡使用OpenAI Whisper的安装教程。
显卡:铭瑄RX580(gfx803架构)
系统:Ubuntu22.04 LTS
1、卸载机器上已有的AMD显卡相关组件
sudo apt autopurge rocm-core amdgpu-dkms
2、安装必要的library
sudo apt install libopenmpi3 libstdc++-12-dev libdnnl-dev ninja-build libopenblas-dev libpng-dev libjpeg-dev
3、添加用户组
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME
4、添加需要的环境变量并重启
sudo sh -c 'echo ROC_ENABLE_PRE_VEGA=1 >> /etc/environment'
sudo sh -c 'echo HSA_OVERRIDE_GFX_VERSION=8.0.3 >> /etc/environment'
sudo reboot
5、AMD显卡组件安装包5.4.3版本
cd ~/Downloads
wget https://repo.radeon.c防删文字om/amdgpu-install/22.40.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
sudo apt install ./amdgpu-install_5.4.50403-1_all.deb
sudo amdgpu-install -y --usecase=rocm,hiplibsdk,mlsdk
6、验证
ls /lib/modules/$(uname -r)/updates/dkms
rocminfo
clinfo
7、下载torch
wget https://ghproxy.c防删文字n/https://github.co防删文字m/tsl0922/pytorch-gfx803/releases/download/pytorch-1.13.1/torch-1.13.1-cp310-cp310-linux_x86_64.whl
下载torchvision
wget https://ghproxy.c防删文字n/https://github.co防删文字m/tsl0922/pytorch-gfx803/releases/download/pytorch-1.13.1/torchvision-0.14.1-cp310-cp310-linux_x86_64.whl
注:如果上述下载链接失效,你可以选择其他github镜像站。当然torch和torchvision也可以自己编译,但本文只提供简单方法。
8、安装torch和torchvision
sudo pip3 install torch-1.13.1-cp310-cp310-linux_x86_64.whl
sudo pip3 install torchvision-0.14.1-cp310-cp310-linux_x86_64.whl
9、添加环境变量
sudo sh -c 'echo LD_LIBRARY_PATH=/opt/rocm-5.4.3/lib/ >> /etc/environment'
10、测试安装是否成功
执行命令python3
然后
>>> import torch
>>> torch.__version__
>>> torch.cuda.is_available()
>>> exit()
如果正确返回版本号及True,说明torch和torchvision安装成功
11、安装ffmpeg和openai-whisper
sudo apt update && sudo apt install ffmpeg
pip3 install -U openai-whisper
12、如果上述过程全都没有报错,OpenAI Whisper就可以在你的Ubuntu22.04上正常使用了。模型会在你首次使用时自动下载。
本教程主要提供整个安装过程,关于OpenAI Whisper的详细使用方法会有另外的文章。
好在AMD的许多软件代码是开源的,官方不支持,开源世界的人们也可以想办法搞定。所以,今天给大家带来RX580显卡使用OpenAI Whisper的安装教程。
显卡:铭瑄RX580(gfx803架构)
系统:Ubuntu22.04 LTS
1、卸载机器上已有的AMD显卡相关组件
sudo apt autopurge rocm-core amdgpu-dkms
2、安装必要的library
sudo apt install libopenmpi3 libstdc++-12-dev libdnnl-dev ninja-build libopenblas-dev libpng-dev libjpeg-dev
3、添加用户组
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME
4、添加需要的环境变量并重启
sudo sh -c 'echo ROC_ENABLE_PRE_VEGA=1 >> /etc/environment'
sudo sh -c 'echo HSA_OVERRIDE_GFX_VERSION=8.0.3 >> /etc/environment'
sudo reboot
5、AMD显卡组件安装包5.4.3版本
cd ~/Downloads
wget https://repo.radeon.c防删文字om/amdgpu-install/22.40.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
sudo apt install ./amdgpu-install_5.4.50403-1_all.deb
sudo amdgpu-install -y --usecase=rocm,hiplibsdk,mlsdk
6、验证
ls /lib/modules/$(uname -r)/updates/dkms
rocminfo
clinfo
7、下载torch
wget https://ghproxy.c防删文字n/https://github.co防删文字m/tsl0922/pytorch-gfx803/releases/download/pytorch-1.13.1/torch-1.13.1-cp310-cp310-linux_x86_64.whl
下载torchvision
wget https://ghproxy.c防删文字n/https://github.co防删文字m/tsl0922/pytorch-gfx803/releases/download/pytorch-1.13.1/torchvision-0.14.1-cp310-cp310-linux_x86_64.whl
注:如果上述下载链接失效,你可以选择其他github镜像站。当然torch和torchvision也可以自己编译,但本文只提供简单方法。
8、安装torch和torchvision
sudo pip3 install torch-1.13.1-cp310-cp310-linux_x86_64.whl
sudo pip3 install torchvision-0.14.1-cp310-cp310-linux_x86_64.whl
9、添加环境变量
sudo sh -c 'echo LD_LIBRARY_PATH=/opt/rocm-5.4.3/lib/ >> /etc/environment'
10、测试安装是否成功
执行命令python3
然后
>>> import torch
>>> torch.__version__
>>> torch.cuda.is_available()
>>> exit()
如果正确返回版本号及True,说明torch和torchvision安装成功
11、安装ffmpeg和openai-whisper
sudo apt update && sudo apt install ffmpeg
pip3 install -U openai-whisper
12、如果上述过程全都没有报错,OpenAI Whisper就可以在你的Ubuntu22.04上正常使用了。模型会在你首次使用时自动下载。
本教程主要提供整个安装过程,关于OpenAI Whisper的详细使用方法会有另外的文章。