Windows10安装Tensorflow2.0

最近开始学习tensorflow,工欲善其事,必先利其器,因此准备安装gpu版本。安装过程相对复杂一点

步骤

  1.确认电脑显卡型号,首先在桌面上右键打开NVIDIA控制面板,再点击左下角的系统信息即可查看显卡相关信息
Windows10安装Tensorflow2.0_第1张图片
Windows10安装Tensorflow2.0_第2张图片
  2.更新显卡驱动,在英伟达官网即可下载,NVIDIA驱动程序下载
如图所示,选择自己的显卡型号以及操作系统,驱动类型在显卡的系统信息内可以看到,如果没有的话就说直接下载标准版,下载完成后直接双击安装即可Windows10安装Tensorflow2.0_第3张图片
  3.安装CUDA,是NVIDIA推出的一种并行计算架构,建议安装CUDA10.0,下载地址为CUDA下载,如图所示,下载完成后,双击安装,默认即可,此时应该需要退出VS,否则无法继续
Windows10安装Tensorflow2.0_第4张图片
安装完成后如何判断是否安装好?
首先打开命令行进入安装路径,如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite,执行deviceQuery.exe,如图
Windows10安装Tensorflow2.0_第5张图片
Windows10安装Tensorflow2.0_第6张图片
如果出现Result = PASS则表示安装成功

  4.下载cuDNN,cuDNN下载,需要注册才能下载,下载界面如下图
Windows10安装Tensorflow2.0_第7张图片
下载完成后,解压,并将cuda中的bin文件夹所处路径加入系统变量,即右键点击此电脑的属性—>高级系统设置—>环境变量—>新建—>复制粘贴bin文件夹路径即可
  5.安装tensorflow,使用pip install tensorflow-gpu -i "https://pypi.doubanio.com/simple/"即可安装,默认版本即为2.0,-i "https://pypi.doubanio.com/simple/"指定豆瓣源下载,避免下载中断(Read timed out)。静待安装完成后,会显示在这里插入图片描述
接下来判断是否安装成功
Windows10安装Tensorflow2.0_第8张图片如果版本号为2.0.0以及is_gpu_available为True则表示安装成功

至此安装Tensorflow2.0的步骤全部结束,我也折腾了两天,一直卡在安装CUDA的那一步,原因就是我安装的时候未退出VS,浪费了很多时间。

你可能感兴趣的:(深度学习)