mirror of
https://github.com/CHN-beta/nixos.git
synced 2024-10-23 05:39:05 +08:00
21 lines
523 B
Nix
21 lines
523 B
Nix
{
|
|
lib, fetchFromGitHub, buildPythonPackage,
|
|
torch, urllib3, requests, cmake, pkg-config, ninja
|
|
}: buildPythonPackage rec
|
|
{
|
|
pname = "torchdata";
|
|
version = "0.7.1";
|
|
src = fetchFromGitHub
|
|
{
|
|
owner = "pytorch";
|
|
repo = "data";
|
|
rev = "v${version}";
|
|
hash = "sha256-SOeu+mI4p2tHX0YyctrDBcrz2/zYcwH9GGJ+6ytRmjQ=";
|
|
fetchSubmodules = true;
|
|
};
|
|
dontUseCmakeConfigure = true;
|
|
pyproject = true;
|
|
propagatedBuildInputs = [ torch urllib3 requests ];
|
|
nativeBuildInputs = [ cmake pkg-config ninja ];
|
|
}
|