【踩坑记录】Tensorflow在Windows下使用

TensorFlow 是一个端到端开源机器学习平台

安装

pip3  install tensorflow

使用时报错如下

2021-04-21 12:04:11.383572: W tensorflow/stream_executor/platform/default/dso_loader.cc:60] Could not load dynamic library 'cudart64_110.dll'; dlerror: cudart64_110.dll not found
2021-04-21 12:04:11.383949: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.

没有 cuDNN64_7.dll 文件，TensorFlow 将无法加载

Windows上搭建TensorFlow的开发环境

需要至少需要安装：

1. python

2. CUDA和CuDNN

CUDA 是NVIDIA发明了一种并行计算平台和编程模型。通过利用图形处理单元（GPU）的功能，可以显着提高计算性能

CUDA下载地址：https://developer.nvidia.com/cuda-90-download-archive

下载后安装

配置环境变量

SET PATH=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\bin;%PATH% SET PATH\=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\extras\\CUPTI\\libx64;%PATH% SET PATH\=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v10.1\\include;%PATH%

查看是否安装成功

nvcc -V

NVIDIA cuDNN是用于深度神经网络的GPU加速的原语库

CuDNN下载地址：https://developer.nvidia.com/cudnn

下载后解压

配置环境变量

SET PATH=C:\\Program Files\\cuda\\bin;%PATH%

查看自己的GPU，配置环境变量

SET PATH=C:\\Program Files\\NVIDIA Corporation\\NVSMI;%PATH%

查看命令

nvidia-smi

3.TensorFlow（GPU版）

安装前一定要确认好每个软件的版本是否相互支持

Bazel 是用于编译 TensorFlow 的构建工具

将 Bazel 可执行文件的位置添加到 %PATH% 环境变量中

下载地址：https://github.com/bazelbuild/bazel/releases

先查看tensorflow版本信息，根据版本对应不同版本的软件

【踩坑记录】Tensorflow在Windows下使用相关推荐

ODrive踩坑（一）windows下使用环境的搭建，odrivetool及USB驱动的安装
最近有空玩玩无刷电机,早就听说ODrive的控制效果不凡,淘宝400买来玩玩.电机使用我以前囤的几个拆机DJI-3512,别看拆机,但悟的电机是针不戳:编码器使用TLE5012B-E1000磁编码器, ...
踩坑记录： Pytorch框架下--- 从零使用卷积神经网络实现人脸面部表情识别 (基于连续维度)
之前一直在自学深度神经网络的知识,在跟着书本一步一步走的时候,感觉每一个思路,每一句代码都特别容易,实现思路清晰明了,实验代码简单易懂.但当我真正课题需要用到的时候,想跳出书本的框架,自行实现并通透其 ...
DJI Windows SDK踩坑记录
DJI Windows SDK踩坑记录前言 DJI Windows SDK现状 DJI官方称WSDK将不会再进行更新维护 DJI Windows SDK无法直接控制Mavic 2 pro的飞行姿态 ...
LinkedIn Camus实践及踩坑记录
LinkedIn Camus踩坑记录首先,介绍下Camus吧. 由LinkedIn公司开发的消息队列同步框架,提供将Kafka(一种消息队列框架)的数据装载到Hadoop分布式文件系统(HDFS)的 ...
Anaconda3安装tensorflow踩坑记录
Anaconda3安装tensorflow2.2踩坑记录本次安装是在py版本为3.6的虚拟环境下安装: 首先激活环境 activate py36(name) activate py36 然后下载 p ...
Python打包工具Pyintealler打包py文件为windows exe文件过程及踩坑记录+实战例子
Python打包工具Pyintealler打包py文件为windows exe文件过程及踩坑记录+实战例子目录 Python打包工具Pyintealler打包py文件为windows exe文件过程 ...
mvn exec: java_实战|Java 测试覆盖率 Jacoco插桩的不同形式总结和踩坑记录(下）
本文为霍格沃兹测试学院优秀学员关于 Jacoco 的小结和踩坑记录.测试开发进阶学习,文末加群. 六.注意事项汇总修改 JAVA_OPTS 参数时,如果位置不对,可能造成代理无法启动. java - ...
VMware WorkStation 16.1.1 安装 Windows 7 虚拟机避免踩坑记录
VMware WorkStation 16.1.1 安装 Windows 7 虚拟机避免踩坑记录安装虚拟机原因宿主机是 Windows 10 LTSC 1809,安装 Virsual Studio ...
配合Opencv2.4.9，CMake3.12.1和VS2010在win10下构建项目踩坑记录
配合Opencv3,CMake和VS2010在win10下构建项目踩坑记录参考https://blog.csdn.net/qq_26623659/article/details/78322782 博 ...

【踩坑记录】Tensorflow在Windows下使用

【踩坑记录】Tensorflow在Windows下使用

【踩坑记录】Tensorflow在Windows下使用相关推荐

最新文章

热门文章