linux - 如何为我的 AMD Linux 系统构建 Tensorflow？

Question

我最近在我的系统上安装了 Tensorflow。拟合模型后，我在 Jupyter Notebook 终端中收到此错误：

This TensorFlow binary is optimized with oneAPI Deep Neural Network Library (oneDNN)to use the following CPU instructions in performance-critical operations:  AVX2 FMA
To enable them in other operations, rebuild TensorFlow with the appropriate compiler flags.

我读到我应该为我的系统构建 Tensorflow，但我不知道如何去做。我需要设置什么样的设置？我需要什么编译器标志？这是我的设置：

显卡：AMD RX 5700XT

CPU：AMD锐龙9 3900X

内存：64GB DDR4

操作系统：Ubuntu 20.04

有人可以就如何为我的系统构建 TensorFlow 提供建议吗？

标签： linuxtensorflowamd

Answer 1

大多数深度学习和机器学习框架和库都使用 NVIDIA CUDA 进行 GPU 处理，我们需要选择 NVIDIA 显卡。

注意虽然 Amd 有一些出色的显卡型号，但它们对 ML 任务的兼容性和支持仍处于试验阶段。所以我们需要坚持使用 NViDia。

AMD 为 AMD GPU 提供了启用 ROCm 的 TensorFlow 库。它基于 ROCm 软件堆栈。要了解有关此库的更多信息：MIOpen - ROCm。

基于这些因素，可以推荐以下显卡系列： GeForce 10 系列 GeForce 16 系列。GeForce 20 系列。GeForce 30 系列