使用支持gpu的tensorflow的前提是安裝了正確版本的CUDA和cuDNN。
關於CUDA和cuDNN的安裝可以參考NVIDIA官網和網上各種安裝教程，在此不再贅述。本文想要強調的重點是要安裝支持自己的GPU的版本，然後根據CUDA版本安裝正確版本的cuDNN，最後根據安裝的CUDA和cuDNN的版本選擇正確的tensorflow版本安裝，否則安裝了tensorflow但是也無法使用GPU，程序跑起來只是在使用CPU。