43 lines
1023 B
Nix
43 lines
1023 B
Nix
{
|
|
description = "A very basic flake";
|
|
|
|
inputs = {
|
|
nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
|
|
flake-utils.url = "github:numtide/flake-utils";
|
|
};
|
|
|
|
outputs = { self, nixpkgs, flake-utils }: flake-utils.lib.eachDefaultSystem (system:
|
|
let pkgs = import nixpkgs {
|
|
inherit system;
|
|
config = {
|
|
allowUnfree = true;
|
|
};
|
|
}; 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 {
|
|
buildInputs = with pkgs; [
|
|
(python3.withPackages(ps: with ps;
|
|
[
|
|
ipython
|
|
jupyter
|
|
numpy
|
|
pandas
|
|
matplotlib
|
|
torch
|
|
scikit-learn
|
|
timeout-decorator
|
|
torchvision
|
|
seaborn
|
|
]))
|
|
];
|
|
};
|
|
}
|
|
);
|
|
}
|