我正在尝试在安装了 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 运行应用程序,而不是重新安装它?
我的天真的方法是创建一个符号链接(symbolic link)......但这是一个可靠的解决方案吗?
或者我可以让 @tensorflow/tfjs-node-gpu
依赖 CUDA 9.1 而不是 9.0 吗?
谢谢!
最佳答案
这是一个已知问题 - 您需要安装 CUDA 9.0:https://github.com/tensorflow/tensorflow/issues/15604
Node.js 上 TensorFlow.js 的底层二进制文件是预先构建的,并且必须与特定版本的 CUDA 链接。请遵循此支持问题,以便构建针对您的系统和库设置进行优化的自定义 libtensorflow:https://github.com/tensorflow/tfjs/issues/571
关于node.js - 我是否必须重新安装 CUDA 9.0 才能使用 tfjs-node-gpu?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54919923/