| 1 | set -ex |
| 2 | |
| 3 | # turn on IBGDA |
| 4 | echo 'options nvidia NVreg_EnableStreamMemOPs=1 NVreg_RegistryDwords="PeerMappingOverride=1;"' | tee -a /etc/modprobe.d/nvidia.conf |
| 5 | |
| 6 | if command -v update-initramfs &> /dev/null; then |
| 7 | # for Debian/Ubuntu |
| 8 | sudo update-initramfs -u |
| 9 | elif command -v dracut &> /dev/null; then |
| 10 | # for Fedora/CentOS |
| 11 | sudo dracut --force |
| 12 | else |
| 13 | echo "No supported initramfs update tool found." |
| 14 | exit 1 |
| 15 | fi |
| 16 | |
| 17 | echo "Please reboot the system to apply the changes" |