add RL to the game (I hope)

This commit is contained in:
Noa Aarts 2025-12-05 20:18:20 +01:00
parent 69710a8e5c
commit 2c93b47630
Signed by: noa
GPG key ID: 1850932741EFF672
3 changed files with 397 additions and 78 deletions

View file

@ -22,10 +22,10 @@
pkgs = import inputs.nixpkgs {
inherit system;
overlays = [ inputs.rust-overlay.overlays.default ];
config = {
allowUnfree = true;
cudaSupport = true;
};
config = {
allowUnfree = true;
cudaSupport = true;
};
};
lib = pkgs.lib;
@ -100,6 +100,7 @@
packages = [
(pkgs.python3.withPackages (ppkgs: [
ppkgs.torch
ppkgs.tqdm
(lib.python_package ppkgs)
]))
];
@ -111,7 +112,7 @@
# To use in other builds with the "withPackages" call
python_package =
ps:
ps.buildPythonPackage rec {
ps.buildPythonPackage {
pname = project_name;
format = "wheel";
version = project_version;