Windows 安装Tensorflow GPU版本


Windows 安装Tensorflow GPU版本

安装CUDA

历史版本

驱动版本要求


安装cuDNN

历史版本

  • 需要下载与CUDA对应的版本

image-20200718171010579

  • 将解压出来的文件夹下的lib, include, bin目录下的文件粘贴到CUDA的文件夹下(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2 | 环境变量PATH中可以看到)

image-20200718181032861

image-20200718181057262


安装tensorflow-gpu

参考

pip3 install --upgrade tensorflow-gpu

可能出现的问题

image-20200718181842958

分析:

  1. 检查安装的Python版本是否为3.6以上的64bit

    • 使用命令python -v查看

      image-20200718182013302

  2. 检查使用的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

文章作者: elegantlee
版权声明: 本博客所有文章除特別声明外,均采用 CC BY 4.0 许可协议。转载请注明来源 elegantlee !
评论
  目录