前言:
应项目需要,要在自己的电脑上安装TensorFlow-GPU版本
若要安装GPU版本则必须要安装CUDA
以下为配合 Win10上搭建TensorFlow(GPU)的学习笔记而做了简单记录.

一、查看显卡支持的CUDA版本号

查看流程如下

  1. 打开控制面板
  2. 硬件和声音
    在这里插入图片描述
  3. NVDIA控制面板
    在这里插入图片描述
  4. 帮助 -> 系统信息
    在这里插入图片描述
  5. 查看支持的CUDA版本号
    在这里插入图片描述

总结
上述栗子,支持的CUDA版本号为10.1.152。

二、CUDA与显卡驱动版本

TensorFlow(GPU) 安装时需要 CUDA+对应的显卡驱动
这里给出英伟达官方的CUDA和驱动的对应:
在这里插入图片描述
新晋小白一枚,文字略显拙劣,有错误欢迎指教~

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐