TensorFlow 2.1启用GPU加速
Python、TensorFlow、NVIDIA GPU驱动、CUDA工具包等软件的版本需要相互匹配,可参考官方的指导“Install TensorFlow 2”与"软件要求"。TensorFlow 2.x已不需独立的TensorFlow-GPU版本。
软件匹配版本:
- Python 3.7
 - NVIDIA GPU驱动 445.87
 - CUDA工具包 10.1
 - cuDNN SDK 7.6
 - TensorFlow 2.1
 
个人博客,记录软件技术与程序员的点点滴滴。
Python、TensorFlow、NVIDIA GPU驱动、CUDA工具包等软件的版本需要相互匹配,可参考官方的指导“Install TensorFlow 2”与"软件要求"。TensorFlow 2.x已不需独立的TensorFlow-GPU版本。
软件匹配版本: