ubuntu 安装 tensorflow
ubuntu 安装 tensorflow
-
查看系统CUDA版本
-
上官网查看CUDA版本对应的tensorflow版本以及python版本:https://www.tensorflow.org/install/source#gpu
需要安装2.15.0版本的tensorflow,并且python版本要在3.9-3.11之间(划重点),python版本对应不上会安装不了对应版本的tensorflow
-
安装tensoorflow:https://www.tensorflow.org/install/pip(注意要看英文版的官网,和中文版的还不一样)
# For GPU users pip install tensorflow[and-cuda] # For CPU users pip install tensorflow
-
验证是否安装成功,gpu是否可用
import tensorflow as tf print(tf.config.list_physical_devices('GPU'))
ubuntu 安装 tensorflow
https://blog.mingchenliu.com/ubuntu 安装 tensorflow/