安装前,先说明自己学习深度学习tensorflow框架,自己买了GTX1070显卡,利用起来。
一、安装python3.7(我选择的,任意)
二、安装CUDA、cudnn
1、从NVIDIA官网(https://developer.nvidia.com/cuda-downloads)下载自己需要的CUDA版本(我选择的10.0),默认一直安装,注意勾选添加到路径。
2、从NVIDIA官网(https://developer.nvidia.com/cudnn)下载自己需要的cudnn版本(要与CUDA版本相匹配),下载下来一个压缩包,解压后,将文件夹里所有文件粘贴到CUDA安装路径下。
三、安装tensorflow-gpu
在线安装方式很多conda、pip都可以
离线安装:强烈建议!
https://pypi.tuna.tsinghua.edu.cn/simple/ 上这个网站找资源,下载好.whl文件。(也就是轮子)
pip install c:\...\tensorflow_gpu-xxxx.whl
四、安装完测试
遇到问题:
1、CUDA与tensorflow版本不匹配,一开始我用的10.1,改成10.0.
tensorflow官网可以看对应版本:https://tensorflow.google.cn/install/source_windows
2、我使用了anaconda,里面的虚拟环境配置学习了。
备注:
https://www.bilibili.com/video/av52282208?from=search&seid=2657292146605146276这个教程在线教安装