在tensorflow的2.4版本中,我们应该在pip安装tensorflow之后单独使用tensorflow gpu吗?

dxpyg8gm  于 2021-07-14  发布在  Java
关注(0)|答案(1)|浏览(545)

我在安装程序中读到,对于2.4版TysFoSurm,安装英伟达CUDA工具包和CUDNN后,就足以完成“PIP安装TunSoFrEx”(访问GPU),而现在不必做“$PIP安装TysFraceGPU”。
我说的对吗?也就是不做“$pip install tensorflow gpu”我们会错过任何功能吗(附上以下图片供您参考)

https://towardsdatascience.com/installing-tensorflow-gpu-in-ubuntu-20-04-4ee3ca4cb75d

qco9c6ql

qco9c6ql1#

如果您安装了CUDNN和英伟达工具箱,那么您可以 pip install tensorflow . 从tensorflow文档(https://www.tensorflow.org/install/gpu):
tensorflow pip包包括对cuda的gpu支持®-启用的卡:
$pip安装tensorflow
本指南介绍最新稳定的tensorflow版本的gpu支持和安装步骤。
tensorflow 1.15及更旧版本的旧版本,cpu和gpu包是分开的:
$pip install tensorflow==1.15#cpu
$pip安装tensorflow gpu==1.15#gpu

相关问题