我的环境
操作系统:windows10
python版本:3.7.3
cuda版本:10.1
cudnn版本:7.6.4
tensorflow-gpu版本:2.0
1.如何安装cuda?
照着从官网下载,直接精简安装就行。你无法选择安装路径,默认c盘。
2.如何安装cudnn?
从官网下载,你会得到一个压缩包,里面有lib,include,bin三个文件夹,把文件夹里面的东西复制到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1这个文件夹里面对应的lib,include,bin文件夹里面去,说的有点拗口,但是就是这么回事。
3.如何安装tensorflow-gpu?
直接命令行敲命令。
pip install tensorflow-gpu
4.你可能面临的问题
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin 这个文件夹下面你可能有 cudart64_101.dll 这个文件,但是你如果运行python代码,会提示你找不到 cudart64_100.dll 这个文件,你可以选择从网上下载,或者直接改名。
还有一个文件要说一下
刚才个文件夹下面你可能有 cublas64_10.dll 这个文件,但是,你需要 cublas64_100.dll 这个东西,我这里是直接改名字了,代码也能运行,就不管那么多了。