mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-09 01:04:57 +08:00
18 lines
477 B
Bash
Executable File
18 lines
477 B
Bash
Executable File
set -ex
|
|
|
|
# turn on IBGDA
|
|
echo 'options nvidia NVreg_EnableStreamMemOPs=1 NVreg_RegistryDwords="PeerMappingOverride=1;"' | tee -a /etc/modprobe.d/nvidia.conf
|
|
|
|
if command -v update-initramfs &> /dev/null; then
|
|
# for Debian/Ubuntu
|
|
sudo update-initramfs -u
|
|
elif command -v dracut &> /dev/null; then
|
|
# for Fedora/CentOS
|
|
sudo dracut --force
|
|
else
|
|
echo "No supported initramfs update tool found."
|
|
exit 1
|
|
fi
|
|
|
|
echo "Please reboot the system to apply the changes"
|