我是否必须重新安装 CUDA 9.0 才能使用 tfjs-node-gpu？答案

【问题标题】：Do I have to reinstall CUDA 9.0 to use tfjs-node-gpu?我是否必须重新安装 CUDA 9.0 才能使用 tfjs-node-gpu？
【发布时间】：2019-02-28 06:48:06
【问题描述】：

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

当我尝试运行应用程序时，我最终会遇到此错误消息，抱怨 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？

谢谢！

【问题讨论】：

你必须安装 CUDA 9.0
好吧，我现在处于一个新的艰难境地。 CUDA 9.0 需要 7.3.0 以外的 GCC，这是 Ubuntu 18.04 中的默认设置

标签： node.js tensorflow tensorflow.js

【解决方案1】：

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

Node.js 上 TensorFlow.js 的底层二进制文件是预构建的，并且必须与特定版本的 CUDA 链接。请在支持上关注此问题，以启用针对您的系统和库设置优化的自定义 libtensorflow：https://github.com/tensorflow/tfjs/issues/571

【讨论】：