Question

我正在尝试在装有ubuntu18.04和nvidia-390的笔记本电脑（Dell xps 15 / GTX 1050）上运行@tensorflow/tfjs-node-gpu示例。

当我尝试运行应用程序时，最终遇到此错误消息，抱怨CUDA的版本。

Error: libcublas.so.9.0: cannot open shared object file: No such file or directory

以前，我已经使用以下命令安装了CUDA工具包：

$ sudo apt-get install nvidia-cuda-toolkit

，并且已安装的CUDA版本是9.1，而不是9.0。

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Nov__3_21:07:56_CDT_2017
Cuda compilation tools, release 9.1, V9.1.85

所以现在我有版本不匹配的问题！我有什么方法可以用CUDA 9.1运行应用程序，而不是重新安装它？

我幼稚的方法是创建符号链接...但这是可靠的解决方案吗？

还是我可以让@tensorflow/tfjs-node-gpu依靠CUDA 9.1而不是9.0？

谢谢！

Answer 1

这是一个已知问题-您需要安装CUDA 9.0：https://github.com/tensorflow/tensorflow/issues/15604

Node.js上TensorFlow.js的基础二进制文件是预先构建的，必须链接到特定版本的CUDA。请遵循以下支持问题，以启用针对您的系统和库设置优化的自定义libtensorflow：https://github.com/tensorflow/tfjs/issues/571

我必须重新安装CUDA 9.0才能使用tfjs-node-gpu吗？

1 个答案: