在ArchLinux上安装GPU版TensorFlow

不得不说,archlinux真的是开发人员友好的…当然前提是能够接受足够新的软件环境(gcc都7.1.1了…)
安装Nvidia驱动,cuda,cudnn,配置环境变量一条命令搞定:

# pacman -S nvidia cuda cudnn

对,你没有看错,就是一条命令,而且以后升级也是,交给系统,完全无忧。

如果不从源码编译的话,直接pip install tensorflow-gpu就好了,That‘s it!

如果从源码编译,请安装bazel,也是一条命令:

# pacman -S bazel

然后参照官方教程编译安装即可。

比Ubuntu简单几个数量级:)

Leave a Reply

Your email address will not be published. Required fields are marked *