为什么要在本地电脑安装 CUDA,CUDA 是什么的,用来做什么?我想,点击标题进来的小伙伴,应该都清楚这些。不管你是用来做什么,或者跟我一样为了跑 Tensorflow 的 Object Detection (目标检测)。
但是我为什么要用 CUDA9,这里版本要特别注意。因为我使用的是 Tensorflow-gpu 1.10.0,为了能运行成功,最好和我的保持一致,那就开始我的踩坑流程吧。
环境:1. windows10 的 64 位电脑
2. cuda9.0.176
3. cudnn7.0.5
CUDA9.01. 下载
1.1. 地址: 略
1.2. 下载流程
2. 安装
2.1. 打开 cuda_9.0.176_win10.exe
刚打开时会有一个漫长的等待过程,然后才会出现如下界面。
2.2. 选择自定义安装
为什么选择自定义,那是因为我开始全部默认直接安装失败。经过网上查询,有些是安装 CUDA 会安装一些本地已经存在的工具,然后又与本地版本冲突等问题。选择自定后会有四个选项,第一先把 NVIDIA GeForce Experience Core 关闭,而 CUDA 下的 Visial Studio Integration 也可以关闭,最后再点到 ”Driver Components“, 如下图上两个可以关闭,而 Display Driver 需要根据下面方法再选择关闭与否。