diff --git a/shell.nix b/shell.nix
new file mode 100644
index 0000000..9347185
--- /dev/null
+++ b/shell.nix
@@ -0,0 +1,12 @@
+with import <nixpkgs> { };
+# Make sure you have CUDA support enabled
+# See: https://nixos.wiki/wiki/Nvidia
+stdenv.mkDerivation {
+  name = "dev";
+  buildInputs = [
+    gcc
+    make
+    opencl-headers
+    cudatoolkit
+  ];
+}