Last active
November 13, 2024 06:31
-
-
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
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# 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 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
libnvrtc.so.12 cannot open shared object file no such file or directory