Skip to content

Instantly share code, notes, and snippets.

@aseaday
Last active November 13, 2024 06:31
Show Gist options
  • Save aseaday/5925dcdda95f6d8d79553336617fb881 to your computer and use it in GitHub Desktop.
Save aseaday/5925dcdda95f6d8d79553336617fb881 to your computer and use it in GitHub Desktop.
Solve nvidia so not found when you use torch or other cuda program
# no import before this
def get_nvidia_lib_paths():
"""Get all NVIDIA library paths for environment setup"""
import os
import nvidia.cublas.lib
import nvidia.cuda_cupti.lib
import nvidia.cuda_nvrtc.lib
import nvidia.cuda_runtime.lib
import nvidia.cudnn.lib
import nvidia.cufft.lib
import nvidia.curand.lib
import nvidia.cusolver.lib
import nvidia.cusparse.lib
import nvidia.nccl.lib
import nvidia.nvjitlink.lib
import nvidia.nvtx.lib
lib_modules = [
nvidia.cublas.lib, nvidia.cuda_cupti.lib, nvidia.cuda_nvrtc.lib, nvidia.cuda_runtime.lib,
nvidia.cudnn.lib, nvidia.cufft.lib, nvidia.curand.lib, nvidia.cusolver.lib, nvidia.cusparse.lib,
nvidia.nccl.lib, nvidia.nvjitlink.lib, nvidia.nvtx.lib
]
return ":".join([os.path.dirname(lib.__file__) for lib in lib_modules + "/lib"])
import os
os.environ["LD_LIBRARY_PATH"] = get_nvidia_lib_paths()
# other imports and code
@aseaday
Copy link
Author

aseaday commented Nov 13, 2024

libnvrtc.so.12 cannot open shared object file no such file or directory

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment