python3Packages.nvidia-dlprof-pytorch-nvtx: init at 1.8.0 (#422437)

{

  lib,

  buildPythonPackage,

  fetchurl,

  pythonOlder,

  numpy,

  torch,

}:

buildPythonPackage rec {

  pname = "nvidia-dlprof-pytorch-nvtx";

  version = "1.8.0";

  src = fetchurl {

    url = "https://pypi.nvidia.com/nvidia-dlprof-pytorch-nvtx/nvidia_dlprof_pytorch_nvtx-${version}-py3-none-any.whl";

    hash = "sha256-IHVCXnkVy3lXw22ISpeMdt5T7UZSzHp8sbWGF/emwGw=";

  };

  format = "wheel";

  dontBuild = true;

  dependencies = [

    numpy

    torch

  ];

  pythonImportsCheck = [

    "nvidia_dlprof_pytorch_nvtx"

  ];

  meta = {

    description = "NVIDIA DLProf Pytorch NVTX markers";

    homepage = "https://docs.nvidia.com/deeplearning/frameworks/dlprof-user-guide/index.html";

    license = lib.licenses.unfree; # NVIDIA Proprietary Software

    maintainers = with lib.maintainers; [ jherland ];

  };

}

  nvdlib = callPackage ../development/python-modules/nvdlib { };

  nvidia-dlprof-pytorch-nvtx =

    callPackage ../development/python-modules/nvidia-dlprof-pytorch-nvtx

      { };

  nvidia-ml-py = callPackage ../development/python-modules/nvidia-ml-py { };

  nwdiag = callPackage ../development/python-modules/nwdiag { };