Win10 64 位Tensorflow-gpu安装(VS2017+CUDA9.2+cuDNN7.1.4+python3.6.5)
Tensorflow 入门系列的第一篇,Tensorflow-gpu的下载与安装。CSDN上能搜到很多学习资料,对初学者非常有帮助。刚安装了目前最新版本的Tensorflow,当然也碰到不少坑,这篇文章详细记录了安装过程,以供参考。 一 本机配置 Win10 64bit (版本1803) AMD Ryzen 5 2400G with Radeon Vega Graphics NVIDIA GeForce GTX 1060 二 文件下载 VS2017 官网社区版 CUDA9.2 官网下载地址 cuDNN7.1 官网下载地址(需要注册) Anacoda5.2 官网下载地址 Python3.6.5 官网下载地址 三 安装过程 VS2017 选择自己想要安装的工作负载,也可以随便选,有需要的时候再修改安装。 详细安装步骤不再赘述,一路默认。 CUDA9.2 选择相对应的版本下载,在线安装版1.5GB,离线安装版14.5MB,先安装基础包,再安装升级补丁。 使用管理员权限安装,一路默认。 安装完成之后打开命令行,输入:nvcc -V查看版本,如果出现以下类似信息表示安装成功。 安装成功后,我的电脑上点右键,打开属性->高级系统设置->环境变量,可以看到系统中多了CUDA_PATH和CUDA_PATH_V9_2两个环境变量。 我们还需要添加如下几个变量: 在用户变量Path下添加: C: