如何在配备 AMD Radeon (TM) 显卡的笔记本电脑上安装 TensorFlow？

Question

我的联想笔记本电脑配备 AMD Radeon (TM) 显卡，我正在尝试安装 TensorFlow，但在编写此代码时总是出现错误：

import os

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"

import tensorflow as tf

physical_devices = tf.config.list_physical_devices("GPU")
tf.config.experimental.set_memory_growth(physical_devices[0], True) 

错误是：

 tf.config.experimental.set_memory_growth(physical_devices[0], True)

 IndexError: list index out of range

如何正确安装tensorflow？

我使用pycharm作为IDE

Answer 1

您可以尝试以下说明在您的笔记本电脑中安装tensorflow...

pip install tensorflow-rocm

然后检查是否安装完美...

import tensorflow as tf
print(tf.__version__)

之后您可以检查您的代码。希望它能成功。