yadunut/nus
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

feat: update flake

Browse Source
This commit is contained in:
Yadunand Prem 2024-04-23 23:41:36 +08:00
parent 89eb024d1e
commit 20b6a552c9
No known key found for this signature in database
1 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
cs2109s/labs/flake.nix
View File

@ -2,7 +2,7 @@
description = "A very basic flake"; description = "A very basic flake";
inputs = { inputs = {
nixpkgs.url = "github:nixos/nixpkgs/nixos-23.11"; nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
flake-utils.url = "github:numtide/flake-utils"; flake-utils.url = "github:numtide/flake-utils";
}; };
@ -11,26 +11,31 @@
inherit system; inherit system;
config = { config = {
allowUnfree = true; allowUnfree = true;
cudaSupport = true;
}; };
}; }; in rec {
in rec { overlays = [(final: prev: {
pythonPackagesExtensions = [(py-final: py-prev: {
torch = py-final.torch-bin;
torchvision = py-final.torchvision-bin;
torchaudio = py-final.torchaudio-bin;
})];
})];
devShell = pkgs.mkShell { devShell = pkgs.mkShell {
buildInputs = with pkgs; [ buildInputs = with pkgs; [
cudaPackages.cudatoolkit (python3.withPackages(ps: with ps;
(python3.withPackages(ps: with ps; [ [
ipython ipython
jupyter jupyter
numpy numpy
pandas pandas
matplotlib matplotlib
torch torch
scikit-learn scikit-learn
timeout-decorator timeout-decorator
torchvision torchvision
seaborn
])) ]))
]; ];
shellHook = "jupyter notebook --ip='100.64.0.8'";
}; };
} }
); );