tensorflow只能在装有gpu的机器上运行

发布网友 发布时间: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。

声明声明:本网页内容为用户发布,旨在传播知识,不代表本网认同其观点,若有侵权等问题请及时与本网联系,我们将在第一时间删除处理。E-MAIL:11247931@qq.com
0.662615s