ubuntu 安装 tensorflow

ubuntu 安装 tensorflow

  1. 查看系统CUDA版本

  2. 上官网查看CUDA版本对应的tensorflow版本以及python版本:https://www.tensorflow.org/install/source#gpu

需要安装2.15.0版本的tensorflow,并且python版本要在3.9-3.11之间(划重点),python版本对应不上会安装不了对应版本的tensorflow

  1. 安装tensoorflow:https://www.tensorflow.org/install/pip(注意要看英文版的官网,和中文版的还不一样)

    # For GPU users
    pip install tensorflow[and-cuda]
    # For CPU users
    pip install tensorflow
  2. 验证是否安装成功,gpu是否可用

    import tensorflow as tf
    print(tf.config.list_physical_devices('GPU'))

ubuntu 安装 tensorflow
https://lmc20020909.github.io/ubuntu 安装 tensorflow/
作者
Liu Mingchen
发布于
2023年12月23日
许可协议