How to Install TensorFlow with GPU Support for AMD on Debian Buster
By Albin / 2020-01-28 — Updated 2021-08-23
The official TensorFlow only supports NVIDIA CUDA-enabled GPUs — which sucks for us AMD users. But there’s a workaround: we’ll use the open source ROCm project from AMD.
1. Add the ROCm Repository and Key
wget -qO - http://repo.radeon.com/rocm/apt/debian/rocm.gpg.key | sudo apt-key add -
echo 'deb [arch=amd64] http://repo.radeon.com/rocm/apt/debian/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list
sudo apt update
2. Check That Your User Is in the video
Group
groups myuser
If not, add yourself:
Read More