前言:
应项目需要,要在自己的电脑上安装TensorFlow-GPU版本
若要安装GPU版本则必须要安装CUDA
以下为配合 Win10上搭建TensorFlow(GPU)的学习笔记而做了简单记录.
一、查看显卡支持的CUDA版本号
查看流程如下:
- 打开控制面板
- 硬件和声音
- NVDIA控制面板
- 帮助 -> 系统信息
- 查看支持的CUDA版本号
总结:
上述栗子,支持的CUDA版本号为10.1.152。
二、CUDA与显卡驱动版本
TensorFlow(GPU) 安装时需要 CUDA+对应的显卡驱动
这里给出英伟达官方的CUDA和驱动的对应:
新晋小白一枚,文字略显拙劣,有错误欢迎指教~