WIN10 + GTX 1060 + Tensorflow GPU版本安装记录

最近买了台本做训练用,因为一直没GPU,今天晚上配置一下TensorFlow环境,配置如下:

  • 安装VS2017

官方下载链接:https://visualstudio.microsoft.com/downloads/?rr=https%3A%2F%2Fwww.baidu.com%2Flink%3Furl%3D7607zWVCfrNBww7i-jC9Idd12bUvZTWfWfQ74KDiDRaVRT9ySKsskOPUcKEByZdTqpRaFQXiNPwQ5-JwXD8nm_%26wd%3D%26eqid%3Dbf2cd077000268f2000000035bf57bdb

百度网盘链接:https://pan.baidu.com/s/1ESJnnxpO2sRey0eE1Am4xA

工作负载选择如下图所示: 

WIN10 + GTX 1060 + Tensorflow GPU版本安装记录_第1张图片

 CUDA版本选择9.0,对应的CUDNN版本为7.4,下面直接给出百度网盘的下载地址。默认安装即可

 百度网盘下载:https://pan.baidu.com/s/1Ak21bYpIPd-H-sjLQ498Dg 

百度网盘下载地址: https://pan.baidu.com/s/1WZkEjTHgTAEKlPNCWDu5GQ

下载解压后(里边有个三个文件)拷贝到cuda9.0文件下 ,如下图所示:

WIN10 + GTX 1060 + Tensorflow GPU版本安装记录_第2张图片

WIN10 + GTX 1060 + Tensorflow GPU版本安装记录_第3张图片

官网下载地址(速度比较慢): https://www.anaconda.com/download/

百度网盘下载地址:https://pan.baidu.com/s/1qDe2mEoFt9cjjQS0B0poDA

安装需注意将Anaconda添加到PATH环境变量中,勾选第一个复选框。

WIN10 + GTX 1060 + Tensorflow GPU版本安装记录_第4张图片

 以管理员身份运行cmd,然后配置清华镜像,输入指令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

 创建运行环境,输入指令:

conda create -n tensorflow-gpu python=3.6

激活并进入环境,使后续指令在激活的环境中生效,输入指令:

conda activate tensorflow-gpu

升级pip到最新版,防止稍后的安装时,出现错误(笔者初次在安装tensorflow时,没有更新pip到最新版,导致下载到一半出现错误),输入指令:

python -m pip install --upgrade pip 

安装TensorFlow GPU版本,输入命令:

pip install tensorflow-gpu

 

你可能感兴趣的:(TensorFlow学习笔记)