Windows 安装Tensorflow GPU版本
安装CUDA
安装cuDNN
- 需要下载与CUDA对应的版本
- 将解压出来的文件夹下的lib, include, bin目录下的文件粘贴到CUDA的文件夹下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2 | 环境变量PATH中可以看到)
安装tensorflow-gpu
pip3 install --upgrade tensorflow-gpu
可能出现的问题
分析:
检查安装的Python版本是否为3.6以上的64bit
使用命令python -v查看
检查使用的pip镜像(亲测华为的镜像不行)
[global] index-url = http://mirrors.aliyun.com/pypi/simple/ trusted-host = mirrors.aliyun.com timeout = 120
目前的环境
CUDA Toolkit: 10.1
cuDNN: v7.6.5
python: 3.7.0 x64
pip: 20.1.1
Pycharm: 2020.1.2 x64
Could not load dynamic library 'cudart64_101.dll'; dlerror: cudart64_101.dll not found