Windows 10安装Cuda9、Cudnn7配置Tensorflow 1.8环境
Windows 10安装Cuda9.0、Cudnn7.0,配置Tensorflow r1.8环境步骤全记录。
当前安装时间:2018年5月10日
安装环境
- Windows 10
- Intel(R) Core(TM) i7-6800K CPU @ 3.40GHz
- 2 * NVIDIA GTX 1080 Ti
- Tensorflow r1.8
- CUDA 9.0、Cudnn 7.0
- Python 3.6
安装CUDA
下载CUDA对应版本:https://developer.nvidia.com/cuda-toolkit-archive,选择**CUDA Toolkit 9.0 (Sept 2017)->Windows->x86_64->10->exe(local)
下载完成后,打开直接点击next下一步进行安装,安装路径默认为C盘
配置Cuda环境变量
- 右击“我的电脑”,选择“属性”,点击“高级系统设置”
- 选择“环境变量”,在“系统变量”框中找到“path”,点击选中,并点击“编辑”按钮
- 点击“新建”按钮,添加以下路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64
- 命令行输入
nccv -V
检查是否成功
安装Cudnn
进入 https://developer.nvidia.com/rdp/cudnn-archive ,选择对应的7.0版本下载即可。
解压压缩包,把压缩包中bin,include,lib中的文件分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0目录的对应目录下。
安装TensorFlow-gpu
1 | pip install tensorflow-gpu |
参考链接
Windows 10安装Cuda9、Cudnn7配置Tensorflow 1.8环境