今天把vscode 更新到v1.62版,试着跑之前写的 ipynb notebook,跑的是tvm相关的代码。
tvm 是自己编译的,依赖的cuda cudnn是自己装的,而不是用conda装的。
jupyter 插件并没有加载自己设定在 ~/.bashrc中的 LD_LIBRARY_PATH 环境变量。
导致 import tvm 会保错,因为找不到依赖的cuda runtime .so文件。
不知道解决,
最后还是在 conda env下装了特定版本的 cuda-runtime ,也不知道还能不能编译模型了,毕竟自己没用conda 装cudnn.
其实 把那些so文件复制到 conda env 下的lib文件应该也行。
vscode 1.62.2版好像没这问题了