1.下载cuda和cudnn
CUDA:https://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html
cuDNN:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#installwindows
官网下载速度慢,已上传,可以直接下
这里我下载的版本是
先下载
链接:https://pan.baidu.com/s/15SRGSEmYU1cemae0KLpM1w
提取码:l6b2
2.安装cuda
参照其他人的文章一步一步下一步基本就可以了,不过值得一提的是安装路径最好默认,这样就直接按照我的教程,如果要安装在其他盘可能要自己注意下我代码路径和你的路径。
https://blog.csdn.net/qq_37296487/article/details/83028394
3.安装CUDNN和配置环境变量
请看这个
https://blog.csdn.net/u010618587/article/details/82940528
这里环境变量我直接写出来你可以复制,我是默认路径
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64
%CUDA_PATH%\bin
%CUDA_PATH%\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
%CUDA_SDK_PATH%\bin\win64
%CUDA_SDK_PATH%\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v9.0
然后在path里编辑再添加
这是我安装成功过后的path里变量,仅供参考,
4.安装TensorFlow
版本兼容问题请看这个
https://blog.csdn.net/qq_27825451/article/details/89082978
安装前打一些补丁
python -m pip install --upgrade pip
pip install --user -U --ignore-installed wrapt enum34 simplejson netaddr
补丁好之后安装gpu版本的TensorFlow
pip install tensorflow-gpu=1.15.0(先安装和配置cuda+cudnn)
5.验证cuda和cudnn安装成功
打开cmd或者anaconda的
进入目录
cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite
输入执行
bandwidthTest.exe
deviceQuery.exe
结果出现
Result = PASS
就代表成功了
5.验证TensorFlow安装成功
import tensorflow as tf