TensorFlow-gpu安装教程(Linux系统）

山山而川_R

已于 2024-04-07 11:50:41 修改

阅读量598

点赞数 2

分类专栏：深度学习各个环境的安装_自用教程文章标签： tensorflow linux 人工智能

于 2024-04-07 11:37:07 首次发布

本文链接：https://blog.csdn.net/m0_60657960/article/details/137457349

版权

8 篇文章 0 订阅

订阅专栏

一、TensorFlow-gpu环境的安装

使用这个代码安装的前提是你的深度学习已经环境存在

（例如：conda、pytorch、cuda、cudnn等环境）

TensorFlow版本对应GPU版本，自己选择版本，也可以忽略版本直接安装TensorFlow-gpu

conda create -n tf3 python==3.8

conda activate tf3

我的cuda环境是11.8、python是3.8、tensorflow-gpu==2.7.0

安装指令：

pip install tensorflow-gpu==2.7.0 -i  https://pypi.mirrors.ustc.edu.cn/simple

python
import tensorflow as tf 
print(tf.__version__)
#输出'2.0.0-alpha0'
print(tf.test.is_gpu_available())
#会输出True,则证明安装成功

python
#新版本的tf把tf.test.is_gpu_available()换成如下命令
import tensorflow as tf 
tf.config.list_physical_devices('GPU')

关注