下载对应版本的cuda,cudNN,anaconda(python3.7)
1、下载cuda_10.0.130_411.31_win10.exe,安装过程默认安装即可,需要注意的是,有时如果你的电脑里有VS,但CUDA提示你需要再次安装,可以在cuda/选项/自定义安装中取消勾选VS,其余过程很简单
2、下载对应版本的cudNN,在cudnn官网上有对应cuda的版本(需要注册账号)
解压之后,得到四个文件,分别为bin,include,lib,**txt
将三个文件夹包含的文件(注意:不是文件夹)放到cuda对应文件夹下
即将:bin中的cudnn64_7.dll 放到 NVIDIA GPUComputing Toolkit/CUDA/v10.0/bin
include中的cudnn.h 放到 NVIDIA GPUComputing Toolkit/CUDA/v10.0/include
lib/x64中的cudnn.lib 放到 NVIDIA GPUComputing Toolkit/CUDA/v10.0/lib/x64
**.txt 放到 NVIDIA GPUComputing Toolkit/CUDA/v10.0中即可
如果版本不对应,在安装tensorflow-gpu会提示你找不到指定文件,例如我先下载了一个10.1版本的cuda,在安装时显示,未找到指定的NVIDIA GPUComputing Toolkit/CUDA/v10.0/bin/cudart64_100.dll。因为我的10.1版本中只有cudart64_101.dll。总之一定要版本对应。
3、anaconda prompt中输入pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.14.0
等待一会,然后就可以安装完成。
tensorflow-gpu安装(win10)
最新推荐文章于 2020-11-22 22:10:00 发布