Fix for #127 - Use common dir for Nvidia Passthrough
This commit is contained in:
parent
1108dbe5be
commit
26747088d5
5
jlmkr.py
5
jlmkr.py
|
@ -365,6 +365,8 @@ def passthrough_nvidia(
|
||||||
.decode()
|
.decode()
|
||||||
.split("\n")
|
.split("\n")
|
||||||
if x
|
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
|
# while mounting the symlink will be resolved and nvidia-smi will appear as a regular file
|
||||||
nvidia_files.add("/usr/bin/nvidia-smi")
|
nvidia_files.add("/usr/bin/nvidia-smi")
|
||||||
|
|
||||||
|
# Use common module dir
|
||||||
|
nvidia_files.add("/usr/lib/x86_64-linux-gnu/nvidia/current")
|
||||||
|
|
||||||
nvidia_mounts = []
|
nvidia_mounts = []
|
||||||
|
|
||||||
for file_path in nvidia_files:
|
for file_path in nvidia_files:
|
||||||
|
|
Loading…
Reference in New Issue