用于 debian 的 Tensorflow GPU

Tensorflow GPU for debian

是否可以在debian中安装tensorflow GPU?我正在使用 Nvidia GTX 1070 ti 和 debian 9.3.0。我已经尝试了几个 Ubuntu 的教程,但都失败了,因为 debian 没有 Ubuntu 支持的相同 PPA 存储库,还看到很多人说不建议将 ubuntu 的存储库添加到 debian

可以,但是比较麻烦:)

我得到了带有 Openbox 的 Debian 9.3,可以很好地与 Tensorflow 1.6 和 Cuda 9.0 + cuDNN 7.0.5.15(最终还有 Wavenet)一起工作。

https://github.com/ella1011/debian_gpu_jungle

您可能要考虑使用 docker/nvidia-docker。 TensorFlow 二进制版本包括 docker 个图像,因此您可以使用它们来避免弄乱您的本地环境。

安装 docker/nvidia-docker 后,它会是这样的:

docker run -it --runtime=nvidia --rm tensorflow/tensorflow:1.6.0-gpu

当然,您可以使用 -v 标志使主机中的目录对 docker 容器可见。