栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

安装TensorFlow-GPU版本,以及CUDA,cduNN

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

安装TensorFlow-GPU版本,以及CUDA,cduNN

查看cuda以及cudnn和tensorflow对应的版本。

从源代码构建  |  TensorFlow

准备安装TensorFlow2.0.0版本的,因为很多书籍和代码都是基于此版本的。

下载CUDA。对应版本Windows 版 CUDA 10.0  

安装

安装的时候注意一下显卡驱动版本,在cmd中输入dxdiag 查看显卡驱动版本,如果CUDA中的显卡驱动版本,比自己电脑带的驱动版本高,就勾选安装,否则不安装。

下载cudnn,cudnn神经加速库。主要是加速并行计算。版本是7.6.5

网址:https://developer.nvidia.com/rdp/cudnn-archive

解压之后,改名

把加速库复制到CUDA10.0中

 接下来配置环境变量,要有三个环境变量 

在anaconda中使用pip安装TensorFlow_gpu==2.0.0

查看是否包含可用的GPU显卡设备

结果1,成功打开cudart64 100.dll库(官网说windows需要这个库)

2. TensorFlow版本2.0.0

3.报错:CUDA的驱动版本不满足CUDA的运行版本,去升级驱动

网址:NVIDIA 驱动程序下载

 下载之后精简安装,精简安装是更新驱动

不要忘了重启电脑,驱动才会生效

使用命令验证是否安装的是TensorFlow GPU版本

import tensorflow as tf
tf.test.is_gpu_available()

 结果显示true, TensorFlow-GPU版安装成功。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/325796.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号