【踩坑记录】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版本信息,根据版本对应不同版本的软件