tensorflow-gpu2.1安装cuda,cudnn,python版本

版本总览

    • 在win10下,安装了tensorflow-gpu因为cuda版本错误了好多,基本都是版本不对引发的。记录一下各组件版本。
      • 直接安装python3.6.8
      • * tensorflow-gpu==2.1
      • * cuda_10.1.105_418.96_win10
      • *cudnn
    • 注意安装cuda时,选择自定义,不要勾选visual studio否则安装失败
      • 解压cudnn,将里面的文件全部分别复制到cuda安装文件下
      • 添加以下四条到系统环境变量的path里
      • 这样就ok啦,如果帮到你,记得点个赞哦!

在win10下,安装了tensorflow-gpu因为cuda版本错误了好多,基本都是版本不对引发的。记录一下各组件版本。

直接安装python3.6.8

下载地址:https://www.python.org/downloads/windows/
找python3.6.8

* tensorflow-gpu==2.1

pip install -i https://pypi.douban.com/simple/ tensorflow-gpu==2.1

* cuda_10.1.105_418.96_win10

cuda_10.1.105_418.96_win10.exe
下载地址:https://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

*cudnn

cudnn-10.1-windows10-x64-v7.6.5.32
下载地址:https://developer.nvidia.com/rdp/cudnn-download

注意安装cuda时,选择自定义,不要勾选visual studio否则安装失败

tensorflow-gpu2.1安装cuda,cudnn,python版本_第1张图片
一直下一步结束就安装成功了

解压cudnn,将里面的文件全部分别复制到cuda安装文件下

cuda文件路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1

添加以下四条到系统环境变量的path里

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

这样就ok啦,如果帮到你,记得点个赞哦!

你可能感兴趣的:(tensorflow)