本指南将帮助您在 Linux 系统上配置 TensorFlow 以使用 GPU 进行加速计算。

系统要求

  • Ubuntu 16.04 或更高版本
  • NVIDIA GPU 和相应的驱动程序

安装步骤

  1. 更新系统
    首先，更新您的系统以获取最新版本：

    sudo apt-get update
sudo apt-get upgrade

  2. 安装 NVIDIA 驱动程序
    下载并安装与您的 GPU 兼容的 NVIDIA 驱动程序。

    sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-<版本号>

    请将 <版本号> 替换为您需要安装的 NVIDIA 驱动程序的版本。

  3. 安装 CUDA Toolkit
    下载并安装 CUDA Toolkit。

    sudo apt-get install cuda

  4. 安装 cuDNN
    下载并安装 cuDNN。

    sudo apt-get install libnccl2 libnccl-dev

  5. 安装 TensorFlow
    使用 pip 安装 TensorFlow。

    pip install tensorflow-gpu

  6. 验证安装
    运行以下命令以验证 TensorFlow 是否正确安装并可以使用 GPU：

    import tensorflow as tf
print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

    如果看到 GPU 的数量，则说明安装成功。

参考资料

如果您需要更多帮助，请访问我们的 TensorFlow 社区论坛。

图片示例

NVIDIA GPU

NVIDIA_GPU

CUDA Toolkit 安装界面

CUDA_Toolkit_Install