使用cuda编译的tensorflow 9.1

Question

我在tensorflow上开始，我想用cuda 9.0安装gpu版本。问题是我的gpu（Nvidia mx150）不能使用cuda 9.0（仅使用cuda 9.1）...所以我尝试使用cuda 9.1编译我自己的tensorflow版本。但是我在编译期间仍然阻止了编译器发现错误因此它没有编译...我不知道为什么但它非常令人沮丧。因此，如果您已经使用cuda 9.1编译了tensorflow版本，我非常感兴趣！

提前谢谢！

Answer 1

几个星期前我遇到了完全相同的问题。

问题是当前版本的TensorFlow（1.7）不支持CUDA9.1，请查看此问题comment以及下面的讨论。

以下是我找到的一些选项：

1. 通过您自己的
从源代码编译TensorFlow
2. 查找现有的whl文件。恩。我使用repo中的whl修正了问题（与Y. Luo的答案相同）
3. 降级至CUDA 9.0
4. 等到Tensorflow支持CUDA9.1：）

Answer 2

如果您不必使用张量流1.7，this repository可能有你想要的。为了清楚起见，我自己从未尝试过任何一种。

如果您需要在Windows上安装，this repository可能会有所帮助。