linux上Tensorflow环境傻瓜式安装方法
深度学习大火,google的Tensorflow是相对来说比较火的框架之一,但是作为小白,配置环境这种事情恶心的不行不行的。
最后发现了一种简单的配置方案,在这里记录一下。
我是使用Xshell在linux服务器中安装的,主要分为以下几步:
- 确认本机的GPU信息
- 安装Anaconda
- 安装对应版本的Tensorflow
为什么说简单呢,因为不用考虑CUDA和cudnn的版本问题,Anaconda会自动读取你的硬件信息进行安装。
- Anaconda的安装:
在本地下载Anaconda安装包,然后上传到服务器,再进行安装。有两个地方需要注意,安装路径记得自己选,是否加入环境变量要选yes。 - 安装Tensorflow:
安装好Anaconda之后运行命令:anaconda search -t conda tensorflow,会出现好多好多可以安装的源,
选一个和你GPU对应的版本,获取下载链接,用命令 anaconda show anaconda/tensorflow(对应的名称),之后你会得到一个链接,再运行conda install 链接-gpu=版本号,就能自动安装了,例如conda install --channel https://conda.anaconda.org/sdvillal tensorflow-gpu=1.3
后面的 -gpu=版本号 很重要,因为如果不加的话会自动安装你能运行的上限,意味着电脑要炸了。所以要选择一个稍微低一些的版本进行安装。
之后就可以愉快的嘿嘿嘿了。