tensorflow-gpu_win10下安装tensorflowGPU

笔记概述

本文记录win10下tensorflow的安装。

由于学习需要，需用到深度学习框架，在此选择tensorflow来进行安装，并记录下win10系统下的安装步骤。本人电脑win10系统，GTX1050Ti显卡，python编程环境为Anaconda

1

确定电脑是否支持CUDA 

主要是看电脑有没有英伟达显卡，显卡支不支持CUDA

2

确定各个tensorflow对应的CUDA/cuDNN版本 

如上图所示，去英伟达官网下载的CUDA/cuDNN一定要对应好版本，不然会导致安装错误，tensorflow使用不了GPU计算.

本文选用CUDA10.0 

cuDNN也要对应：

本文采用cuDNN下载地址：

https://developer.nvidia.com/rdp/cudnn-download

CUDA下载地址：

https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

j第一步是安装CUDA，安装完后解压cuDNN,并将里面的文件复制到CUDA安装目录下

3

Anaconda创建python3.6编程环境安装tf

具体创建可以参考之前的笔记Anaconda下创建不同版本的Python环境

利用pip进行安装：

pip install tensorflow-gpu

要更快可以使用国内源进行下载，可参考笔记pip使用国内源安装模块

在运行过程中可能会提示缺少cudart64_100.dll文件等，这时解决方案是到网上下载这个文件并进行替换。

测试：

import tensorflow as tftf.test.is_gpu_available()

上面代码返回True则说明支持GPU运算了

挨踢的土木佬欢迎来踢