windows下安装tensorflow-GPU (快速搭建版)

我的电脑配置  
显卡:GTX 950M
CPU:i7-4700HQ
注:只有NVIDIA显卡才能使用 

**

一、 anaconda的安装

**
如果有梯子的话可以在官网下载
anaconda官网:https://www.anaconda.com/distribution/
没有的话在这里给出百度云链接:
链接:https://pan.baidu.com/s/1kMEAbg00T8I6BieGQWfYGA
提取码:slhj

双击安装就行了
安装教程请去搜下;

**

二、 anaconda创建环境

**

(1) 打开 anaconda prompt 如下
在这里插入图片描述
(2)输入conda create -n mytensorflow python=3.6
(mytensorflow指虚拟环境名称 , python=3.6使用python3.6作为环境基础)
在这里插入图片描述
(3) 激活环境

	(base) C:\Users\Administrator>activate mytensorflow

**

三、 在虚拟环境中安装tensorflow-gpu

**
安装gpu版本tensorflow,这里使用conda install tensorflow-gpu
(使用pip安装的话会各种报错,好像是anaconda中有些包和pip不兼容),安装比较慢静心等待就好了

**

四、测试是否安装成功

**
在命令行输入

(mytensorflow ) C:\Users\Administrator>python 
//进入python命令行
>>import  tensorflow as tf
//引入包

如果不报错的话就安装成功了

**

五、总结

**
tensorflow-GPU与tensorflow-CPU快了10多倍,加快了模型的训练效率。
这里不推荐单独安装
CUDA、cudnn、tensorflow-GPU因为会出现好多版本不兼容问题耽误时间,还不如直接用anaconda集成环境来搭建 ,安装方便卸载都很方便。

发布了14 篇原创文章 · 获赞 8 · 访问量 6139

猜你喜欢

转载自blog.csdn.net/KangTongShun/article/details/90734917