前言:
根据项目需要,在自己的电脑上安装TensorFlow-GPU版本
必须安装CUDA才能安装GPU版本
以下是根据Win10上搭建TensorFlow(GPU)的学习笔记简要记录的。
一、显卡支持的CUDA版本号查看流程如下:
打开控制面板上的硬件和声音
NVDIA控制面板
帮助-系统信息
支持的CUDA版本号总结:
上述板栗支持的CUDA版本号为10.1.152。
二、CUDA和显卡驱动程序版本TensorFlow(GPU)安装时CUDA+对应的显卡驱动
以下是NVIDIA官方的CUDA和驱动程序的对应关系。新晋朴素的哑铃一枚,文字略显拙劣,有错误欢迎指教~