1、查看本机的CUDA驱动适配版本

桌面右键打开英伟达控制面板,点击帮助->系统信息->组件
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

tensorflow各个版本与cuda版本的对应关系

参考https://tensorflow.google.cn/install/source_windows#gpu

根据参考,我的电脑就是安装tensorflow_gpu-2.0.0版本,CUDA安装10.0版本,cudnn是7.4版本

2、 安装CUDA

下载地址:https://developer.nvidia.com/downloads(注意:国内访问很卡,需要借助特殊工具)
在这里插入图片描述
在这里插入图片描述
下图拉到最下面
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

下载完成后安装cuda

选择解压路径, 注 意 : 这 里 是 解 压 路 径 , 不 是 安 装 路 径 , 不 要 和 安 装 路 径 是 相 同 的 , 我 开 始 \color{red}{注意:这里是解压路径,不是安装路径,不要和安装路径是相同的,我开始}
以 为 是 安 装 路 径 , 和 后 面 的 安 装 路 径 是 一 样 的 , 导 致 安 装 完 成 之 后 就 没 了 \color{red}{以为是安装路径,和后面的安装路径是一样的,导致安装完成之后就没了}
在这里插入图片描述

等待安装完成
在这里插入图片描述
安装完成后,等待检查系统兼容性
在这里插入图片描述
检查完成后,选择同意并继续
在这里插入图片描述
选择自定义安装方式
在这里插入图片描述
先把NVIDIA GeForce Experience 的对勾去掉,不安装它
在这里插入图片描述
如果电脑没有安装Visual Studio 那就把Visual Studio Intergration 勾去掉,即便安装了也可以取消
在这里插入图片描述
查看驱动版本,如果当前版本大于新版本,那就取消掉勾,我这里取消掉了
在这里插入图片描述
下面的由于我的当前版本和新版本相同,我就取消勾选了,也可以不取消
在这里插入图片描述

最后点击下一步安装,然后选择安装位置,默认就可以了,继续下一步,等待大概10分钟
安装完成之后关闭
在这里插入图片描述

关闭之后找到软件安装目录,比如我的如下图,找到nvcc.exe文件,确认有就可以
在这里插入图片描述
再找到如下文件,确认有就可以
在这里插入图片描述

3、安装cuDNN

下载地址:https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
选择window10的下载
在这里插入图片描述
下载需要先注册为英伟达账号,直接用提示的社交网络账号登陆就行,需要验证邮箱
注 意 : 不 要 使 用 q q 邮 箱 , 等 了 半 天 没 消 息 , 很 是 墨 迹 , 很 慢 ! ! ! 换 了 163 的 立 马 发 了 验 证 邮 件 ! \color{red}{注意:不要使用qq邮箱,等了半天没消息,很是墨迹,很慢!!!换了163的立马发了验证邮件! } 使qq163
下载完成之后,解压
在这里插入图片描述
解压之后文件夹名字是cuda
在这里插入图片描述
改名为cudnn
在这里插入图片描述

然后复制整个文件夹到CUDA安装目录下,比如我的是:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0,然后确认一下cudnn文件夹下是否有这个文件
在这里插入图片描述

4、添加环境变量

到系统环境变量里德Path里,确认一下是否有一下两个,安装CUDA的时候会默认添加
在这里插入图片描述
然后添加CUPTI环境变量
在这里插入图片描述
添加完成之后上移到第三的位置
在这里插入图片描述
在这里插入图片描述
同样的把cudnn/bin目录添加到环境变量中,并且上移到第四位置
在这里插入图片描述
保存环境变量后,打开控制台,输入:nvcc -V 查看版本信息
在这里插入图片描述

5、安装tensorflow

自行安装anaconda

创建虚拟环境

conda create -n tf-gpu python=3.6

激活环境:

activate tf-gpu

安装tensorflow

pip install --upgrade tensorflow-gpu==2.0.0

进入python,查看tensorflow版本
在这里插入图片描述
查看一下gpu可否使用
在这里插入图片描述
Anaconda的使用参见我的另一篇博文:https://blog.csdn.net/s_daqing/article/details/117846862

GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2 PiperOrigin-RevId: 663726708 2 个月前
91dac11a This test overrides disabled_backends, dropping the default value in the process. PiperOrigin-RevId: 663711155 2 个月前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐