需要做的事情:
1、下载安装Anaconda3 python3.7版本
2、通过pip 安装tensorflow_gpu
3、查看自己的安装的tensorflow对应的cuda 和cudnn的版本
4、下载安装cuda 和cudnn
5、安装visual Studio2017
1、Anaconda3的安装
Anaconda3下载地址:https://www.anaconda.com/distribution/
安装:
当有多个用户的选择 All users
安装位置:把路径改为 D盘
两个都钩上,第一个是加入环境变量的
Anaconda的环境配置。
此电脑>>属性>>>高级系统设置>>>环境变量>>>系统变量
>>PATH
2、pip 安装tensorflow_gpu
win+R >>>cmd pip install tensorflow_gpu 等待安装完毕
有时需要升级pip : python -m pip install --upgrade pip
3、安装cuda 和cudnn
安装好tensorflow后查看自己需要安装对应的cuda和cudnn版本。
D: \Python\ Anaconda \Lib\site-packages\tensorflow\ python\platform\ build_info.py
打开build_info.py这个文件。
先安装CUDA,在安装cudnn
CUDA10.0的下载安装:
下载地址: https://developer.nvidia.com/cuda-toolkit-archive
安装在默认位置就可以
CUDA的环境配置
安装结束后是否安装成功测试:
nvcc -V
4、下载cudnn
1、下载地址: https://developer.nvidia.com/rdp/cudnn-download
2、把cudnn的文件复制到CUDA中
解压cudnn压缩包,把压缩包中bin,include,lib中的文件分别拷贝C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0目录下对应目录
5、成功导入tensorflow
1、查看自己安装tensorflow的版本
pip list 查看已安装的tensorflow
2、导入tensorflw包
import tensorflow