my Docker is build from FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
In my local build , I can find libcuda.so.1 and the the code runs successfully
aicrowd@hc-A100-server:/submission$ find /usr/ -name libcuda.so.1
/usr/local/cuda-12.1/compat/libcuda.so.1
/usr/lib/x86_64-linux-gnu/libcuda.so.1
when I submmit , I got a ERROR
ImportError: libcuda.so.1: cannot open shared object file: No such file or directory
I have try this code, but it does not work
import os
def find_libcuda():
paths_to_search = [‘/usr/lib’, ‘/usr/local/lib’, ‘/usr/lib/x86_64-linux-gnu’] # 添加可能的路径
for path in paths_to_search:
full_path = os.path.join(path, ‘libcuda.so.1’)
if os.path.exists(full_path):
return full_path
return None
libcuda_path = find_libcuda()
if libcuda_path:
print(f’Found libcuda.so.1 at: {libcuda_path}')
else:
raise FileNotFoundError(‘libcuda.so.1 not found’)
os.environ[“LD_LIBRARY_PATH”] = f"{os.path.dirname(libcuda_path)}:" + os.environ.get(“LD_LIBRARY_PATH”, “”)
os.environ[“CUDA_VISIBLE_DEVICES”] = “0,1,2,3”
Could you please give me some help. THANKS