Fix for #127 - Use common dir for Nvidia Passthrough

This commit is contained in:
Darren Gibbard 2024-05-09 10:48:55 +01:00
parent 1108dbe5be
commit 26747088d5
1 changed files with 6 additions and 1 deletions

View File

@ -365,6 +365,8 @@ def passthrough_nvidia(
.decode()
.split("\n")
if x
# Strip individual modules for current kernel, since we add the whole directory later
and "/usr/lib/x86_64-linux-gnu/nvidia/current" not in x
]
)
)
@ -382,6 +384,9 @@ def passthrough_nvidia(
# while mounting the symlink will be resolved and nvidia-smi will appear as a regular file
nvidia_files.add("/usr/bin/nvidia-smi")
# Use common module dir
nvidia_files.add("/usr/lib/x86_64-linux-gnu/nvidia/current")
nvidia_mounts = []
for file_path in nvidia_files: