(绕坑)搭建tensorflow-gpu运行环境,cuda+Anaconda+python+tensorflow
作者本人是想利用Tensorflow进行VGG-16框架训练,一开始用的CPU,训练起来心态,都可以烤牛排,就想搭载GPU版的,在搭建tensorflow-gpu的时候,由于一开始装了个python版本,在Anaconda上面创建虚拟环境的时候老是会出现混乱。所以建议大家就仅用Anaconda,挺好用的。
个人建议:注意版本选择,以及你自己的驱动(cuda8.0+cudnn6.0+Anaconda-python3.7+tensorflow-gpu-1.1)
由于前面几次失败经历,给出一套比较稳定的选择,同时tensorflow不要太新版的,因为2.0舍弃了一些函数,创建VGG-16模型的时候会出现警报(个人强迫症!!!!不喜欢)
虽然说Anaconda是python3.7的但是没关系,我们可以创建一个虚拟环境(用python3.6)
旧版本下载:
Cudnn:https://developer.nvidia.com/rdp/cudnn-archive
Cuda:https://developer.nvidia.com/cuda-toolkit-archive
下载完成后安装上,网上很多教程就不一一例举了,挺简单的,但是版本一定要对应好,这是重点!
下载安装Anaconda-py-3.7,直接去官网下载就ok,去清华软件镜像站下载也可以(推荐,因为下载比较快,附上链接):https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
安装过程也不多说了,网上都有,但是安装Anoconda会占用比较大的空间,如果小伙伴们的C盘很紧凑的话建议其他盘
安装完毕后应该有这些,点(红圈)这个,
进去,然后建议点击Channels加一个清华的镜像站链接,因为安装的时候会快一些,我加的有两个,给大家贴出来
1、https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
2、https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
创建环境咯!我已经创建一个了tf-gpu的,但是没影响
点击create,选择py版本(个人建议3.6),然后创建就ok了QAQ
然后点击这个
进去,进入到那个环境!activate tf-gpu (这是你们自己命名的,可能有区别)
我们python一下,可以看到版本,然后用 aconda install tensorflow-gpu 命令安装tensorflow-gpu
然后'y'就ok了,就会自动安装咯!(我就不安装示范了,因为我有一个环境了)
然后检查一下是否安装好!键入,如果出现类似界面,恭喜你配置完成!!!!!
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
然后如何在pycharm里面使用这个环境呢?
打开pycharm
File->setting->project:recognizer-project->project interpreter->add
然后添加,添加python.exe就可以了,确定保存设置
我们来检验一下!我们可以看到现在用的就是Anaconda配置的环境咯!
(小建议:如果小伙伴想要看到训练进度,可以不用pycharm,,可以用Spyder,但是你要切换环境再用哦!!!)
喜欢的小伙伴可以来点点赞QAQ~~~~