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(注意要看英文版的官网,和中文版的还不一样)
-
验证是否安装成功,gpu是否可用
ubuntu 安装 tensorflow
https://lmc20020909.github.io/ubuntu 安装 tensorflow/