Instalando o ROCm 3.3.0 no Ubuntu 18.04 para Deep Learning
O ROCm (Radeon Open Compute) é a plataforma da AMD para computação de alto desempenho (HPC) e aprendizado profundo (Deep Learning), permitindo que usuários aproveitem o potencial de GPUs AMD para tarefas intensas. Este tutorial mostra como instalar o ROCm 3.3.0 no Ubuntu 18.04, utilizando apenas comandos seguros e seguindo boas práticas de sistema.

Pré-requisitos
- Ubuntu 18.04 com kernel 4.17 ou superior.
- CPU moderna com suporte a PCIe Gen3 e PCIe Atomics.
- GPU AMD compatível com ROCm.
CPUs compatíveis com ROCm
- AMD Ryzen, Threadripper, EPYC
- Intel Xeon E3/E5/E7 v3+
- Intel Core i3/i5/i7 de 4ª geração (Haswell ou superior)
- Alguns sistemas Ivy Bridge-E
GPUs compatíveis
- GFX8: Radeon R9 Fury X, Radeon Instinct MI8
- GFX9: Radeon RX Vega 64, Radeon Instinct MI25/MI50/MI60, Radeon VII
⚠️ Para a lista mais recente e oficial, consulte o repositório oficial do ROCm no GitHub: https://github.com/RadeonOpenCompute/ROCm
Passo a passo de instalação
1. Atualize o sistema e instale dependências
sudo apt update
sudo apt dist-upgrade
sudo apt install libnuma-dev
sudo reboot
2. Adicione o repositório ROCm
wget -q -O - https://repo.radeon.com/rocm/apt/3.1/rocm.gpg.key | sudo apt-key add -
echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/3.1/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list
sudo apt update
⚠️ Possíveis erros
Você pode encontrar mensagens como:
W: Distribuição em conflito: https://repo.radeon.com/rocm/apt/debian ubuntu InRelease (esperado ubuntu mas obtido focal)
Esses erros geralmente indicam inconsistência no arquivo sources.list
. Certifique-se de estar usando a URL e release corretos (como xenial
para Ubuntu 18.04).
Solução sugerida
sudo apt clean
sudo apt update
3. Instale o ROCm
sudo apt install rocm-dkms
4. Conceda permissão de acesso à GPU
sudo usermod -a -G video $LOGNAME
5. Teste a instalação
/opt/rocm/bin/rocminfo
/opt/rocm/opencl/bin/x86_64/clinfo
Esses comandos devem retornar informações sobre sua GPU AMD.
6. Adicione o ROCm ao PATH
echo 'export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin/x86_64' | sudo tee -a /etc/profile.d/rocm.sh
Instalando TensorFlow com suporte a ROCm
1. Instale bibliotecas essenciais
sudo apt install rocm-libs miopen-hip cxlactivitylogger rccl
sudo apt install wget python3-pip
2. Instale o TensorFlow
pip3 install --user tensorflow-rocm
Isso instalará o TensorFlow já otimizado para trabalhar com ROCm em sua GPU AMD.
Conclusão
Com o ROCm e o TensorFlow devidamente instalados, seu Ubuntu 18.04 está preparado para aproveitar o potencial das GPUs AMD em projetos de aprendizado profundo. Essa solução é uma alternativa econômica ao uso de GPUs NVIDIA, mantendo performance e escalabilidade.
💡 Sempre consulte as documentações oficiais para atualizações e compatibilidade com novos drivers ou versões:
Autor: Seu Nome
Data de publicação: 27 de setembro de 2023
#rocm; #deeplearning; #ubuntu1804