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

Posted 2023-03-15

参考技术A 是的。
使用支持gpu的tensorflow的前提是安装了正确版本的CUDA和cuDNN。
关于CUDA和cuDNN的安装可以参考NVIDIA官网和网上各种安装教程，在此不再赘述。本文想要强调的重点是要安装支持自己的GPU的版本，然后根据CUDA版本安装正确版本的cuDNN，最后根据安装的CUDA和cuDNN的版本选择正确的tensorflow版本安装，否则安装了tensorflow但是也无法使用GPU，程序跑起来只是在使用CPU。