如何用cuda8.0安装TensorFlow-gpu?

时间:2018-02-02 04:52:03

标签: tensorflow tensorflow-serving

我尝试按照官方网站上的说明安装它,导致tensorflow导致导入错误:

<!DOCTYPE html>
<html lang="en">
<head>
  <meta charset="UTF-8">
  <title>Title</title>


</head>
<body>

<div id="box">
  <input type="radio" v-model="pick" :value="a">
</div>

<script type="text/javascript" src="https://cdn.bootcss.com/vue/2.5.13/vue.min.js"></script>
<script type="text/javascript" src="https://cdn.bootcss.com/vue-resource/1.3.4/vue-resource.min.js"></script>
<script type="text/javascript" src="index.js"></script>
</body>
</html>

我运行代码ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory ,这表明我的cuda版本是8.0.61。

似乎tensorflow正在寻找cuda 9.0。我无法升级cuda,因为我正在使用共享的gpu-server而且我没有root权限。

有没有办法让张量流与cuda 8.0一起使用?或者其他任何方式?

谢谢!

2 个答案:

答案 0 :(得分:35)

您需要为CUDA-8安装public class MyTabLayout extends TabLayout { public MyTabLayout(Context context, AttributeSet attrs) { super(context, attrs); int normalColor = ContextCompat.getColor(context, R.color.tabTextColor); int selectedColor = ContextCompat.getColor(context, R.color.tabSelectedTextColor); setTabTextColors(normalColor, selectedColor); } } 版本

1.4.1

最新版本(版本pip install tensorflow-gpu==1.4.1 )适用于CUDA-9

答案 1 :(得分:11)

我遇到了类似的问题,直到找到

https://www.tensorflow.org/install/install_sources#tested_source_configurations

检查已安装的cuda版本和cudnn版本,然后找出哪个版本的tensorflow-gpu与使用上述链接的版本兼容。

我已经安装了cuda 8和cudnn v5.1,因此通过检查以上链接tensorflow-gpu 1.2.0是否兼容以及在使用来安装后,

pip install tensorflow-gpu==1.2.0

对我有用。