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版本。
软件匹配版本: