【环境搭建】win10安装TensorFlow全过程(python+anaconda+pycharm+CPU版本+GPU版本)

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_41683218/article/details/81220804

笔记本配置:

windows10

I7 CPU

显卡940MX

一、安装python

1.从官网上下载windows安装包,下载地址

记得勾选,将python加入环境变量 。如果读者不希望把python安装到C盘,可以点击Customize installation,跟换安装路径。

二、Anaconda安装

1.下载安装包

2.双击打开安装包

3.同意

4.可以选择修改安装的路径,博主默认安装在C盘

5.添加环境变量,第二个勾可选可不选。

6.安装需要一段时间,耐心等待。 

7.安装完后检查是否成功安装

按win+R,输入cmd

8.输入

conda --version

如图,则成功安装。

三、安装TensorFlow(CPU版本) 

1.开始菜单->Anaconda3—>Anaconda Prompt(或者win+r输入cmd) :

 2.输入命令,创建环境

conda create --name tensorflow python=3.5

3.输入Y按回车

4.等待一段时间,环境就会创建成功

5.激活环境

activate tensorflow 

6.安装TensorFlow

pip install --upgrade --ignore-installed tensorflow

7.测试是否成功

win+R输入cmd打开命令窗口,输入python

依次输入以下代码

    import tensorflow as tf
    s = tf.constant('ok, TensorFlow!')
    sess = tf.Session()
    print(sess.run(s))

8.安装成功 

四、安装pycharm

1.下载安装包,这里我们下载社区版的,免费使用。

2.安装过程,无非就是一直next

 

3.创建一个新的工程或者按config 

 

5.修改环境

 

pycharm到此就安装成功。

五、安装tensorflow-gpu版本

1.下载安装包

cudnn-9.0-windows10-x64-v7.1 下载

cuda_9.0.176_win10 下载

(博主经常下载完会断线,所以后来上传到了百度云盘,下载地址

Visual Studio 2015 下载

ps:CUDA和cudnn的版本一定要一致!!!不能一个9.1一个9.0!!!cudnn下载需要登录,用微信就行了。

2.安装包准备完

3.安装Visual Studio 2015

VS安装过程比较久,大概需要1-2个小时

4.安装CUDA

安装成功

5.安装cudnn 

将cudnn解压,得到

然后将3个文件夹复制到C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0 路径下

6.添加系统环境变量

先检查一下CUDA_PATH和CUDA_PATH_V9.0已经存在。

手动添加 C:\ProgramData\NVIDIA GPU Computing Toolkit\v9.0\bin 到path里面

 

7.安装tensorflow的GPU版本

按win+R输入cmd

然后输入

pip install tensorflow-gpu

由于我已经安装过了,所以不会有安装过程。大家安装的时候,要等待一段时间。

8.检验

打开python环境

输入以下代码

 import tensorflow as tf
 s = tf.constant('ok, TensorFlow-gpu!')
 sess = tf.Session()
 print(sess.run(s))

若出现下图,则成功。

940MX是博主的显卡。

展开阅读全文

没有更多推荐了,返回首页