Azure Virtual Machines
An Azure service that is used to provision Windows and Linux virtual machines.
8,173 questions
To use Tensorflow on AMD GPUs, you can use ROCm.
How to use Tensorflow with AMD Vega 10 GPU
![](data:image/svg+xml, %3Csvg xmlns='http://www.w3.org/2000/svg' height='64' class='font-weight-bold' style='font: 600 30.11764705882353px "SegoeUI", Arial' width='64'%3E%3Ccircle fill='hsl(25.6, 56.00000000000001%, 12%)' cx='32' cy='32' r='32' /%3E%3Ctext x='50%25' y='55%25' dominant-baseline='middle' text-anchor='middle' fill='%23FFF' %3EME%3C/text%3E%3C/svg%3E)
Mike Evans
0
Reputation points
Hello,
I have been using Azure virtual machines to train deep learning models in Tensorflow. Recently, I created an NVas v4 machine with the Data Science VM Ubuntu image and am unable to use the GPU. My understanding is that the NVas v4 GPU is an AMD Vega 10 (as indicated by the output of lshw -class display). Therefore, because the GPU is not made by NVIDIA the usual drivers and libraries that Tensorflow would use to recognize the GPU (CUDA, cuDNN) are irrelevant. Are there additional packages/libraries needed to use Tensorflow with an AMD GPU? It seems odd that Microsoft would offer VMs with GPUs that do not have the necessary software to be utilized.
Mike
1 answer
Sort by: Most helpful
-
Vahid Ghafarpour 22,290 Reputation points2023-10-20T17:44:56.42+00:00 -
Mike Evans 0 Reputation points
2023-10-20T23:06:03.0133333+00:00 Thanks for the insight @Vahid Ghafarpour I recently came across upon ROCm. Have you had success installing on Azure VMs? I have been following the package manager installation instructions https://rocm.docs.amd.com/en/latest/deploy/linux/index.htmlwithout success. Specifically,
sudo apt install rocm-hip-sdkfails due to unmet dependencies. -
Vahid Ghafarpour 22,290 Reputation points
2023-10-21T02:42:55.3866667+00:00 I do not have experience installing it on Azure VM, but I hope this thread can help you.
https://github.com/RadeonOpenCompute/ROCm/issues/1713
I think this command can help you based on the thread
amdgpu-install --usecase=opencl,graphics -y
Sign in to comment -