window10安装tensorflow-gpu
admin
11月 09, 2020
0
Window10安装tensorflow-gpu
版本介绍
- python 3.5.4
- tensorflow-gpu 1.12.0
- CUDA 9.0
- CUDNN 7.4.1.5
安装步骤
- 默认已经安装过python3.5.x
- 安装vs2015 https://www.microsoft.com/zh-cn/download/details.aspx?id=48145
- 下载cuda 9.0版本
- 下载cudnn 7.4.1.5版本
- 安装cuda 9.0 ,默认选项安装即可
- 解压cudnn 7.4.1.5
- 将相应的文件拷贝到相应的目录中
- copy
- cudnnbincudnn64_5.dll -> C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0bin
- cudnnincludecudnn.h -> C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0include
- cudnnlibx64cudnn.lib -> C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0libx64
- 将C:Program FilesNVIDIA GPU Computing ToolkitCUDAv8.0bin添加到系统环境变量(path)中
- 检查cuda+cudnn是否安装正确
- 打开命令行
- 输入nvcc -V
- 看到版本信息即为成功
- 安装virtualenv
- 创建虚拟环境
- 进入虚拟环境
- 安装tensorflow-gpu 这里版本是1.12.0
- pip install tensorflow-gpu
- 测试是否安装正确
- import tensorflow as tf
- 没有报错就是安装成功
近期评论