【转载】win10下安装GPU版本的TensorFlow(cuda + cudnn)

链接:https://cloud.tencent.com/developer/article/1475455

一、 明确自己电脑的适配版本的cuda

打开NVIDIA控制面板——>设置physx配置——>组件,可以看到NVIDIA.DLL, 后面的就是你应该安装的版本

二、下载对应版本的cuda, cudnn

cuda下载地址:https://developer.nvidia.com/cuda-toolkit-archive

cudnn下载地址:https://developer.nvidia.com/rdp/cudnn-download(在下载时需要先注册)

三、安装cuda, cudnn

cuda正常安装

解压与cuda对应的cudnn,将全部文件移到 你的cuda安装目录下 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

其中v10.0为cuda安装版本,以个人实际下载版本为准

四、检验CUDA,CUDNN是否安装成功

1.win+R打开cmd——>输入nvidia-smi——>nvcc -V
2.cd到根目录——>在 cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite——>分别输入deviceQuery.exe 和 bandwidthTest.exe,显示Result=PASS即安装成功

五、安装TensorFlow-GPU版本

在anaconda下安装ensorflow-gpu

©️2020 CSDN 皮肤主题: 数字20 设计师:CSDN官方博客 返回首页