最近开始学习tensorflow,工欲善其事,必先利其器,因此准备安装gpu版本。安装过程相对复杂一点
步骤
1.确认电脑显卡型号,首先在桌面上右键打开NVIDIA控制面板,再点击左下角的系统信息即可查看显卡相关信息
2.更新显卡驱动,在英伟达官网即可下载,NVIDIA驱动程序下载
如图所示,选择自己的显卡型号以及操作系统,驱动类型在显卡的系统信息内可以看到,如果没有的话就说直接下载标准版,下载完成后直接双击安装即可
3.安装CUDA,是NVIDIA推出的一种并行计算架构,建议安装CUDA10.0,下载地址为CUDA下载,如图所示,下载完成后,双击安装,默认即可,此时应该需要退出VS,否则无法继续
安装完成后如何判断是否安装好?
首先打开命令行进入安装路径,如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite,执行deviceQuery.exe,如图
如果出现Result = PASS则表示安装成功
4.下载cuDNN,cuDNN下载,需要注册才能下载,下载界面如下图
下载完成后,解压,并将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)。静待安装完成后,会显示
接下来判断是否安装成功如果版本号为2.0.0以及is_gpu_available为True则表示安装成功
至此安装Tensorflow2.0的步骤全部结束,我也折腾了两天,一直卡在安装CUDA的那一步,原因就是我安装的时候未退出VS,浪费了很多时间。