ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

tensorflow安装

2021-01-14 02:02:11  阅读:19  来源: 互联网

标签:cuDNN cudnn CUDA NVIDIA tensorflow 安装 cuda


1.cuda安装

https://developer.nvidia.com/cuda-10.1-download-archive-base

运行exe文件

点击同意

选择自定义

cuda选择中取消visual studio

如果当前版本驱动程序大于cuda驱动程序版本则不安装,等于随便。小于的话一定安装。

接下来一路默认即可

点击关闭

检查环境变量

测试

2.cudnn安装

打开网址 https://developer.nvidia.com/cudnn,选择“Download cuDNN”,由于 NVIDIA公司的规定,下载 cuDNN 需要先登录,因此用户需要登录或创建新用户后才能继续下载。登录后,进入 cuDNN 下载界面,勾选“I Agree To the Terms of the cuDNN SoftwareLicense Agreement”,即可弹出 cuDNN 版本下载选项。选择 CUDA 10.1 匹配的 cuDNN 版本,并点击“cuDNN Library for Windows 10”链接即可下载 cuDNN 文件 。需要注意的是,cuDNN 本身具有一个版本号,同时它还需要和 CUDA 的版本号匹配上,不能下错不匹配 CUDA 版本号的 cuDNN 文件。  下载下来以后是一个zip文件,解压即可 将文件夹复制到这个路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1并重命名为cudnn 添加两条环境变量往path中:  C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\cudnn\bin 并将其置顶 3。安装tensflow-gpu 对应关系       执行以上安装命令。 4.测试(返回true就成功了)

标签:cuDNN,cudnn,CUDA,NVIDIA,tensorflow,安装,cuda
来源: https://www.cnblogs.com/jinshilong/p/14275224.html

专注分享技术,共同学习,共同进步。侵权联系[admin#icode9.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有