【求助】taichi无法调用cuda后端问题

综合讨论区
#1

求助各位巨佬，我最近在服务器端掉用taichi时候使用"ti.init(arch=ti.cuda)"出现报错

[Taichi] version 1.6.0, llvm 15.0.4, commit f1c6fbbd, linux, python 3.8.16
[E 07/21/23 16:17:19.912 1637593] [dynamic_loader.cpp:load_function@40] Cannot load function: /lib/x86_64-linux-gnu/libcuda.so: undefined symbol: cuDeviceGetDefaultMemPool

[W 07/21/23 16:17:19.913 1637593] RuntimeError: ‘[dynamic_loader.cpp:load_function@40] Cannot load function: /lib/x86_64-linux-gnu/libcuda.so: undefined symbol: cuDeviceGetDefaultMemPool’ occurred when detecting cuda, consider adding TI_ENABLE_CUDA=0 to environment variables to suppress this warning message.
[W 07/21/23 16:17:19.913 1637593] [misc.py:adaptive_arch_select@753] Arch=[<Arch.cuda: 4>] is not supported, falling back to CPU
[Taichi] Starting on arch=x64

我的服务器配置是
Linux VM-0-4-ubuntu 5.4.0-139-generic #156-Ubuntu SMP x86_64 x86_64 x86_64 GNU/Linux
cuda安装正常，能够使用torch
不知道这是什么样的错误？万分感谢！

#2

你好:wave: 这个看起来是 CUDA 版本比较旧的问题，如果可以的话升级一下 CUDA driver 的版本。另外为了帮助进一步锁定问题，你能否提供一下在终端输入 nvidia-smi 的信息？

#3

截屏2023-07-23 13.09.29
截屏2023-07-23 13.09.29810×144 6.54 KB

您好！我的driver版本是这样的。我觉得您说的很有道理，但是服务器的driver版本我没有权限改动，所以我降级了taichi到1.4，发现现在可以正常调用了！
感谢您！

1 个赞