python3Packages.{torchaudio,torchvision,vllm}: use torch.stdenv

pkgs/development/python-modules/torch/default.nix

···
     echo
     echo "# Update using: unroll-src [version]"
   '';
+
+  stdenv' = if cudaSupport then cudaPackages.backendStdenv else stdenv;
 in
 buildPythonPackage rec {
   pname = "torch";
   # Don't forget to update torch-bin to the same version.
   version = "2.6.0";
   pyproject = true;
+
+  stdenv = stdenv';
 
   outputs = [
     "out" # output standard python package

pkgs/development/python-modules/torchaudio/default.nix

···
   version = "2.6.0";
   pyproject = true;
 
+  stdenv = torch.stdenv;
+
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "audio";

pkgs/development/python-modules/torchvision/default.nix

···
 
 let
   inherit (torch) cudaCapabilities cudaPackages cudaSupport;
-  inherit (cudaPackages) backendStdenv;
 
   pname = "torchvision";
   version = "0.21.0";
 in
 buildPythonPackage {
   inherit pname version;
+
+  stdenv = torch.stdenv;
 
   src = fetchFromGitHub {
     owner = "pytorch";
···
       export TORCHVISION_INCLUDE="${libjpeg_turbo.dev}/include/"
       export TORCHVISION_LIBRARY="${libjpeg_turbo}/lib/"
     ''
-    # NOTE: We essentially override the compilers provided by stdenv because we don't have a hook
-    #   for cudaPackages to swap in compilers supported by NVCC.
     + lib.optionalString cudaSupport ''
-      export CC=${backendStdenv.cc}/bin/cc
-      export CXX=${backendStdenv.cc}/bin/c++
       export TORCH_CUDA_ARCH_LIST="${lib.concatStringsSep ";" cudaCapabilities}"
       export FORCE_CUDA=1
     '';

pkgs/development/python-modules/vllm/default.nix

···
   depyf,
   opencv-python-headless,
 
-  config,
-
-  cudaSupport ? config.cudaSupport,
+  cudaSupport ? torch.cudaSupport,
   cudaPackages ? { },
-  rocmSupport ? config.rocmSupport,
+  rocmSupport ? torch.rocmSupport,
   rocmPackages ? { },
   gpuTargets ? [ ],
-}@args:
+}:
 
 let
   inherit (lib)
···
 
   cpuSupport = !cudaSupport && !rocmSupport;
 
-  # https://github.com/pytorch/pytorch/blob/v2.4.0/torch/utils/cpp_extension.py#L1953
+  # https://github.com/pytorch/pytorch/blob/v2.6.0/torch/utils/cpp_extension.py#L2046-L2048
   supportedTorchCudaCapabilities =
     let
       real = [
···
         "8.9"
         "9.0"
         "9.0a"
+        "10.0"
       ];
       ptx = lists.map (x: "${x}+PTX") real;
     in
···
   version = "0.7.3";
   pyproject = true;
 
-  stdenv = if cudaSupport then cudaPackages.backendStdenv else args.stdenv;
+  stdenv = torch.stdenv;
 
   src = fetchFromGitHub {
     owner = "vllm-project";