From ee07d15d5307b6009a486351c04335bf03fa7d90 Mon Sep 17 00:00:00 2001
From: Adnan Wahab <shel.bernstein123@gmail.com>
Date: Wed, 15 May 2024 21:42:02 -0700
Subject: [PATCH] ok

---
 python.nix | 34 ++++++++++++++++++++++++++++++++++
 1 file changed, 34 insertions(+)
 create mode 100644 python.nix

diff --git a/python.nix b/python.nix
new file mode 100644
index 00000000..21c7cce3
--- /dev/null
+++ b/python.nix
@@ -0,0 +1,34 @@
+{ config, lib, pkgs, ... }:
+   {
+      description = "Python 3.9 development environment";
+      outputs = { self, nixpkgs }:
+      let
+        system = "aarch64-linux";
+        pkgs = import nixpkgs {
+          inherit system;
+          config.allowUnfree = true;
+          config.cudaSupport = true;
+        };
+      in {
+        devShells.${system}.default = pkgs.mkShell {
+          buildInputs = with pkgs; [
+            cudatoolkit linuxPackages.nvidia_x11
+            cudaPackages.cudnn
+            libGLU libGL
+            xorg.libXi xorg.libXmu freeglut
+            xorg.libXext xorg.libX11 xorg.libXv xorg.libXrandr zlib
+            ncurses5 stdenv.cc binutils
+            ffmpeg
+            python39
+            python39Packages.pip
+            python39Packages.numpy
+            python39Packages.pytorch-bin
+            python39Packages.virtualenv
+          ];
+
+          shellHook = ''
+              export LD_LIBRARY_PATH="${pkgs.linuxPackages.nvidia_x11}/lib"
+          '';
+        };
+      };
+    }