diff --git a/pkgs/development/python-modules/nvidia-dlprof-pytorch-nvtx/default.nix b/pkgs/development/python-modules/nvidia-dlprof-pytorch-nvtx/default.nix
new file mode 100644
index 000000000000..72b9bdb5072e
--- /dev/null
+++ b/pkgs/development/python-modules/nvidia-dlprof-pytorch-nvtx/default.nix
@@ -0,0 +1,38 @@
+{
+  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 ];
+  };
+}
diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix
index 5d3262b216a4..2d60c4fac66f 100644
--- a/pkgs/top-level/python-packages.nix
+++ b/pkgs/top-level/python-packages.nix
@@ -10420,6 +10420,10 @@ self: super: with self; {
 
   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 { };