Windows10下配置TensorFlow-GPU的详细过程
导读
参考了一些文章,挣扎了半天把这个环境配好了,分享一下过程。
三步骤
1. 安装anaconda
2. 安装cuda和cudnn
3. 下载TensorFlow-gpu和测试
说明: 本人的显卡:GeForce GTX 750 Ti
1. 安装anaconda
官网下载:https://www.anaconda.com/distribution
选择Windows下Python3.7(注意:必须选择64位,因为TF不支持Python32位的)
下载完成后打开,然后就是傻瓜式的安装,一路next即可。
然后打开anaconda prompt命令界面,按照下面的代码创建一个虚拟环境
conda create tf-2.0 # 创建环境
activate tf-2.0 # 激活环境
2. 安装cuda 和cudnn
1)安装cuda
目前TensorFlow-GPU版最高只能用cuda-10.0的。
下载链接:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
下载完成后,打开下载的驱动。安装路径默认就行。
取勾GeForce Experience
如果电脑上本身就有Visual Studio Integration,要将这个取消勾选,避免冲突了
点开Driver comonents,Display Driver这一行,前面显示的是Cuda本身包含的驱动版本是411.31
如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号,那一定要把这个勾去掉。否则会安装失败(相同的话,就不用去取勾了)(其实这个失败不失败我不知道,本人装的时候是取消了。参考另外一篇文章的,链接在末尾)
接下来需要几分钟的时间安装,这是NVIDIA程序已完成的界面
打开此路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
,查看是否有nvcc.exe。有就说明CUDA安装已成功。
2) 安装cudnn
CDNN官网链接:
https://developer.nvidia.com/rdp/cudnn-download
选择cuDNN for CUDA10.0
需要注册一个账号,并且验证成功。网很重!!!
解压cuDNN到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
3)配置路径
查看CUDA环境路径
我的电脑——>属性——>高级系统设置——>环境变量
在系统变量中找到Path,双击打开添加两个路径:一是cudnn的bin路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn\bin
,二是C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v10.0\extras\CUPTI\libx64
4) 测试成功与否
cmd进入命令提示符输入:nvcc - V
(大写)
3. 下载TensorFlow-gpu和测试
1)下载TensorFlow-GPU
cmd打开命令提示符:
[1] activate tf-2.0
[2] pip install tensorflow-gpu==2.0.0-beta0
[3] pip install keras
等待安装完毕。
2) 测试
进入python(tf-2.0)环境(在命令提示符找到前面创建的环境),比如本人的创建路径:
进入之后,输入以下代码,验证是否安装成功:
import tensorflow as tf
tf.test.is_gpu_available()
如图显示说明安装成功:
更好的表述是生活一部分