如何解决tensorflow2.x版本无法调用GPU的问题?
- 内容介绍
- 文章标签
- 相关推荐
本文共计1223个文字,预计阅读时间需要5分钟。
最近学校提供了一个服务器账号,用于训练神级网络。服务器配置了十路Titan V,安装了TensorFlow 2.2。Python版本是3.6.2。安装完毕后,用tf.test.is_gpu_available()检查能否调用GPU。
最近学校给了一个服务器账号用来训练神经网络使用,服务器本身配置是十路titan V,然后在上面装了tensorflow2.2,对应的python版本是3.6.2,装好之后用tf.test.is_gpu_available()查看是否能调用gpu,结果返回结果是false,具体如下:
这里tensorflow应该是检测出了gpu,但是因为某些库无法打开而导致tensorflow无法调用,返回了false,详细查看错误信息可以看到一行:
可以看到上面几个文件都顺利打开了,但是最后一个libcudnn.so.7文件显示无法打开,not such file or directory。怀疑是cuda有问题,服务器本身是装的cuda10.1,跟tensorflow2.2应该是吻合的,但是一直无法调用,所以一开始我想重新安装cuda,覆盖掉服务器本来的cuda,下好安装包之后,因为我不是管理员,没有root权限,所以总是失败。
本文共计1223个文字,预计阅读时间需要5分钟。
最近学校提供了一个服务器账号,用于训练神级网络。服务器配置了十路Titan V,安装了TensorFlow 2.2。Python版本是3.6.2。安装完毕后,用tf.test.is_gpu_available()检查能否调用GPU。
最近学校给了一个服务器账号用来训练神经网络使用,服务器本身配置是十路titan V,然后在上面装了tensorflow2.2,对应的python版本是3.6.2,装好之后用tf.test.is_gpu_available()查看是否能调用gpu,结果返回结果是false,具体如下:
这里tensorflow应该是检测出了gpu,但是因为某些库无法打开而导致tensorflow无法调用,返回了false,详细查看错误信息可以看到一行:
可以看到上面几个文件都顺利打开了,但是最后一个libcudnn.so.7文件显示无法打开,not such file or directory。怀疑是cuda有问题,服务器本身是装的cuda10.1,跟tensorflow2.2应该是吻合的,但是一直无法调用,所以一开始我想重新安装cuda,覆盖掉服务器本来的cuda,下好安装包之后,因为我不是管理员,没有root权限,所以总是失败。

