add RL to the game (I hope)
This commit is contained in:
parent
69710a8e5c
commit
2c93b47630
3 changed files with 397 additions and 78 deletions
11
flake.nix
11
flake.nix
|
|
@ -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;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue