Win10+Tensorflow+keras+GPU环境配置

配置:
Windows 10专业版(版本10.0.15063)
Intel Core i3-3420 @ 3.4GHz
GeForce GTX 1060 6GB
RAM 16GB

需要文件:
vs2015.3.pro_chs.iso
cuda_8.0.61_win10.exe
cudnn-8.0-windows10-x64-v5.0-ga.zip
Anaconda3-4.2.0-Windows-x86_64.exe

1,安装VS2015或2013

2,安装CUDA8.0

配置环境变量

CUDA_HOME:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

添加path变量:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\extras\CUPTI\libx64

3,安装cudnn

解压cudnn-8.0-windows10-x64-v5.0-ga.zip

复制粘贴lib,include,bin文件夹到

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下

4,安装Anaconda

5,安装tensorfow

conda create -n tensorflow 
activate tensorflow
pip install --ignore-installed --upgrade  https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-    1.1.0-cp35-cp35m-win_amd64.wh

6,安装keras

pip install keras

7,我装的是VS2013(安装速度比2015快太多)装完后import tensorflow 没有显示成功加载CUDA库,但是使用的时候,执行会话使用的是GPU。所以不影响使用。

img

img

如果文章对您有用请随意打赏,谢谢支持!
0%