安装前的话
首先,我是极力不推荐在windows环境下安装tensorflow的,特别是那些想用gpu加速的童鞋,因为一个不好你就踩进坑了,并且花费大量时间精力去爬坑,但是linux下就轻松多了
先看看我们要装什么东西:
- tensorflow-gpu(如果只用cpu训练,就装tensorflow)
- numpy
- scipy
- cudnn(装tensorflow的童鞋不用装)
- cudatoolkit(装tensorflow的童鞋不用装)
- miniconda
安装conda
conda这种东西可谓是懒人神器,替你把需要安装的依赖包都想好了,miniconda是anaconda的瘦身版,搜一下就有下载地址,搜不到的童鞋可以去清华镜像源下。
我是用windows远程连接linux机器进行训练的,pycharm的专业版中需要指定解释器的地址,注意地址是在linux机器的minconda/scrips文件夹下的python.exe,相应的配置教程网上有。
安装tensorflow
接下来就是创建env,然后source activate env_name,这样激活你的环境后就可以进行一键下载啦(我是直接安装在root环境下的,别学我,懒),输入命令 conda install tensorflow-gpu 确认后就会自动下载所有需要的包,如果网速太慢又没有科学上网条件的童鞋,直接切换成镜像源去下载,速度起飞。
注意事项
有些linux系统cudnn是不支持的,大家记得去nvdia官网看下到底支持哪些版本,但是ubuntu、centos、pandora之类的还是支持的。