2 files changed, 15 insertions, 0 deletions

diff --git a/scripts/default.nix b/scripts/default.nix
index 220d76d..f134556 100644
--- a/scripts/default.nix
+++ b/scripts/default.nix
@@ -36,6 +36,9 @@ let
   # battery script
   battery = import ./battery.nix pkgs;
 
+  # run-on-gpu script
+  battery = import ./nvidia-offload.nix pkgs;
+
   # fzf script to switch between tmux sessions
   tmux-fzf = pkgs.writeScriptBin "tmux-fzf" ''
     LIST_DATA="#{session_name}/#{window_name}/#{pane_current_command} @ #{pane_current_path}"
diff --git a/scripts/nvidia-offload.nix b/scripts/nvidia-offload.nix
new file mode 100644
index 0000000..de55fa2
--- /dev/null
+++ b/scripts/nvidia-offload.nix
@@ -0,0 +1,12 @@
+{ pkgs, ... }:
+
+let
+  name = "nvidia-offload";
+in
+pkgs.writeShellScriptBin name ''
+  export __NV_PRIME_RENDER_OFFLOAD=1
+  export __NV_PRIME_RENDER_OFFLOAD_PROVIDER=NVIDIA-G0
+  export __GLX_VENDOR_LIBRARY_NAME=nvidia
+  export __VK_LAYER_NV_optimus=NVIDIA_only
+  exec "$@"
+''