如何为我的AMD Linux系统构建Tensorflow?

如何为我的AMD Linux系统构建Tensorflow?,linux,tensorflow,amd,Linux,Tensorflow,Amd,我最近在我的系统上安装了Tensorflow。在安装到模型后,我在Jupyter笔记本终端中发现以下错误: 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 oper

我最近在我的系统上安装了Tensorflow。在安装到模型后,我在Jupyter笔记本终端中发现以下错误:

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,但我不知道如何构建。我需要设置什么样的设置?我需要什么编译器标志?以下是我的设置:

GPU:AMD RX 5700XT

CPU:AMD Ryzen 9 3900X

内存:64GB DDR4

操作系统:Ubuntu 20.04


有人能给我一些关于如何为我的系统构建Tensorflow的建议吗?

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

请注意,虽然Amd有一些优秀的图形卡型号,但它们与ML任务的兼容性和支持仍处于试验阶段。所以我们需要坚持使用英伟达

AMD为AMD GPU提供一个支持ROCm的TensorFlow库。它基于ROCm软件栈。了解更多关于这个图书馆的信息:MIOpen-ROCm

基于这些因素,可以推荐以下图形卡系列: GeForce 10系列 GeForce 16系列。 GeForce 20系列。
GeForce 30系列

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

请注意,虽然Amd有一些优秀的图形卡型号,但它们与ML任务的兼容性和支持仍处于试验阶段。所以我们需要坚持使用英伟达

AMD为AMD GPU提供一个支持ROCm的TensorFlow库。它基于ROCm软件栈。了解更多关于这个图书馆的信息:MIOpen-ROCm

基于这些因素,可以推荐以下图形卡系列: GeForce 10系列 GeForce 16系列。 GeForce 20系列。 GeForce 30系列

看看这个,也许会派上用场看看这个,也许会派上用场