diff --git a/pkgs/development/python-modules/tritonclient/default.nix b/pkgs/development/python-modules/tritonclient/default.nix
index 47c87866aad1..49393e0c7671 100644
--- a/pkgs/development/python-modules/tritonclient/default.nix
+++ b/pkgs/development/python-modules/tritonclient/default.nix
@@ -1,43 +1,49 @@
 {
   lib,
-  fetchPypi,
+  stdenv,
   buildPythonPackage,
-  pythonOlder,
-  python-rapidjson,
+  fetchPypi,
   numpy,
+  python-rapidjson,
   # optional dependencies
-  grpcio,
-  packaging,
   aiohttp,
   geventhttpclient,
+  grpcio,
+  packaging,
 }:
 
 let
   pname = "tritonclient";
-  version = "2.35.0";
-in
-buildPythonPackage {
-  inherit pname version;
+  version = "2.54.0";
   format = "wheel";
+in
+buildPythonPackage rec {
+  inherit pname version format;
 
-  disabled = pythonOlder "3.7";
-
-  src = fetchPypi {
-    inherit pname version;
-    python = "py3";
-    dist = "py3";
-    format = "wheel";
-    platform = "manylinux1_x86_64";
-    hash = "sha256-JXVr+DWkM2g7CiU9STMcZyQJieXgNbKXlX/jqf/oam8=";
-  };
+  src =
+    let
+      platforms = {
+        aarch64-linux = "manylinux2014_aarch64";
+        x86_64-linux = "manylinux1_x86_64";
+      };
+      hashes = {
+        aarch64-linux = "e485a67c75121a2b58456bd6275086252dd72208674b0c85bd57a60f428b686f";
+        x86_64-linux = "53c326498e9036f99347a938d52abd819743e957223edec31ae3c9681e5a6065";
+      };
+    in
+    fetchPypi {
+      inherit pname version format;
+      python = "py3";
+      dist = "py3";
+      platform = platforms.${stdenv.hostPlatform.system} or { };
+      sha256 = hashes.${stdenv.hostPlatform.system} or { };
+    };
 
   propagatedBuildInputs = [
-    python-rapidjson
     numpy
+    python-rapidjson
   ];
 
-  doCheck = false;
-
   pythonImportsCheck = [ "tritonclient" ];
 
   passthru = {
@@ -54,7 +60,7 @@ buildPythonPackage {
   };
 
   meta = with lib; {
-    description = "Triton python client";
+    description = "Triton Python client";
     homepage = "https://github.com/triton-inference-server/client";
     license = licenses.bsd3;
     maintainers = with maintainers; [ happysalada ];