随着硬(you)件(xi)更新换代越来越快,在这记录安装 tensorflowgpu 支持的过程概要
安装过程
-
确定你电脑的 GPU 型号,只支持绿卡。
-
安装 anaconda(可选)
-
pip 安装 tensoflow-gpu,这一步关系到下一步的版本号
-
确定 GPU 型号所适配的 CUDA 版本,与 cuDNN 版本
这里讲解一下,CUDA 相当于显卡的并行计算专用驱动,cuDNN 相当于程序需要的 API
根据 tensorflow 包版本确定需要的 CUDA 版本和 cuDNN 版本
下载指定版本的 CUDA 版本包
先注册登录,然后下载指定版本的 cuDNN -
开始安装 CUDA 和 cuDNN
安装 CUDA 时建议选自定义,驱动必选,因为它所需的驱动版本比你一般玩游戏需要的都要新。然后将 VS 和它的一个依赖去掉,不然会又大又慢
如果安装好了,可以在 cmd 尝试输入nvcc -V
cuDNN 可以不叫安装,解压后可以直接覆盖到 CUDA 的安装目录下(bin,include,lib/x64),下一步的环境变量就比较好配置
-
配置系统环境变量 Path
可能是直接配好的,也可能需要你添加一下,具体路径仅供参考(CUDA 和 cuDNN 放在一起的情况下)
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
欢迎来到这里!
我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。
注册 关于