如何在Linux上安装张量流

don*_*lan 4 python tensorflow

在TensorFlow主网站上,您必须专门启用gpu:

# Ubuntu/Linux 64-bit, GPU enabled:
$ sudo pip3 install --upgrade https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.7.1-cp34-none-linux_x86_64.whl
Run Code Online (Sandbox Code Playgroud)

在anaconda网站上,您只需要安装张量流,而没有指示这是GPU还是CPU版本:

conda install -c https://conda.anaconda.org/jjhelmus tensorflow
Run Code Online (Sandbox Code Playgroud)

运行Anaconda命令后,我决定需要确保它是GPU版本,因此我尝试在anaconda文件夹中运行pip命令。

当我运行pip命令时,出现以下错误:

tensorflow-0.7.1-cp34-none-linux_x86_64.whl在此平台上不受支持


我不知道轮子是什么...但是我有一台新的笔记本电脑,配备多核i7(去年7月)和笔记本电脑nvidia卡,安装了5.2的计算等级和CUDA(Ubuntu 15.10)。

必须有一个与我的规格兼容的版本。

如何将张量流安装升级到GPU版本?

更新资料

当我运行:时import pip; print(pip.pep425tags.get_supported()),支持的轮子包括

cp35

但是TensorFlow可以使用的轮子是cp34 ...那么这是否意味着GPU上的TensorFlow对我的计算机不起作用?

bna*_*aul 5

看起来很愚蠢,只要将轮子从3.4重命名为3.5就足够了:

$ wget https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.7.1-cp34-none-linux_x86_64.whl
...
$ pip install tensorflow-0.7.1-cp34-none-linux_x86_64.whl
tensorflow-0.7.1-cp34-none-linux_x86_64.whl is not a supported wheel on this platform.
$ mv tensorflow-0.7.1-cp3{4,5}-none-linux_x86_64.whl
$ pip install tensorflow-0.7.1-cp35-none-linux_x86_64.whl
Processing ./tensorflow-0.7.1-cp35-none-linux_x86_64.whl
...
Installing collected packages: tensorflow
Successfully installed tensorflow-0.7.1
Run Code Online (Sandbox Code Playgroud)