Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.3 kB · 70 lines
Nix
at litex
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071{ lib, buildPythonPackage, fetchFromGitHub, pythonOlder, fsspec, lightning-utilities, numpy, packaging, pyyaml, tensorboardx, torch, torchmetrics, tqdm, traitlets
# tests, psutil, pytestCheckHook}:
buildPythonPackage rec { pname = "pytorch-lightning"; version = "2.0.2"; format = "pyproject";
src = fetchFromGitHub { owner = "Lightning-AI"; repo = "pytorch-lightning"; rev = "refs/tags/${version}"; hash = "sha256-MSztKWjg/7J+4+sv4sqFlucaYuQlGoehtcUTiqNUlPA="; };
preConfigure = '' export PACKAGE_NAME=pytorch '';
propagatedBuildInputs = [ fsspec numpy packaging pyyaml tensorboardx torch lightning-utilities torchmetrics tqdm traitlets ] ++ fsspec.optional-dependencies.http;
nativeCheckInputs = [ psutil pytestCheckHook ];
# Some packages are not in NixPkgs; other tests try to build distributed # models, which doesn't work in the sandbox. doCheck = false;
pythonImportsCheck = [ "pytorch_lightning" ];
meta = with lib; { description = "Lightweight PyTorch wrapper for machine learning researchers"; homepage = "https://pytorch-lightning.readthedocs.io"; license = licenses.asl20; maintainers = with maintainers; [ tbenst ]; };}