本文目的主要是为了介绍如何在win10环境下安装tensorflow,适合gpu显卡型号为1070,python使用版本是3.6的电脑
1 准备
(1) vs2015 下载
(2) cuda 9.1下载
(3) cuDNN 下载
(4) 显卡驱动下载
2 VS2015安装
从头到尾默认安装
3 cuda环境配置
(1) 默认安装cuda_9.1.85_win10
(2) 默认安装cuda_9.1.85.1_windows
(3) 默认安装cuda_9.1.85.2_windows
(4) 默认安装cuda_9.1.85.3_windows
(5) 配置环境变量
使用nvcc -V在PowerShell查看是否安装成功
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:08:12_Central_Daylight_Time_2017
Cuda compilation tools, release 9.1, V9.1.85
4 更新显卡驱动
默认安装419.35-desktop-win10-64bit-international-whql
5 tensorflow 安装
使用conda安装gpu版本的tensorflow,这里不建议采用pip进行安装,使用pip安装不能解决tensorflow-gpu的安装依赖问题