发布网友 发布时间:2022-04-21 22:43
共1个回答
热心网友 时间:2023-09-18 01:58
是的。
使用支持gpu的tensorflow的前提是安装了正确版本的CUDA和cuDNN。
关于CUDA和cuDNN的安装可以参考NVIDIA官网和网上各种安装教程,在此不再赘述。本文想要强调的重点是要安装支持自己的GPU的版本,然后根据CUDA版本安装正确版本的cuDNN,最后根据安装的CUDA和cuDNN的版本选择正确的tensorflow版本安装,否则安装了tensorflow但是也无法使用GPU,程序跑起来只是在使用CPU。