diff --git a/pkgs/development/python-modules/torchvision/default.nix b/pkgs/development/python-modules/torchvision/default.nix index 7ec3e7858b56..2c81797f2e91 100644 --- a/pkgs/development/python-modules/torchvision/default.nix +++ b/pkgs/development/python-modules/torchvision/default.nix @@ -1,7 +1,10 @@ { lib, + stdenv, torch, + apple-sdk_13, buildPythonPackage, + darwinMinVersionHook, fetchFromGitHub, # nativeBuildInputs @@ -44,11 +47,20 @@ buildPythonPackage { which ] ++ lib.optionals cudaSupport [ cudaPackages.cuda_nvcc ]; - buildInputs = [ - libjpeg_turbo - libpng - torch.cxxdev - ]; + buildInputs = + [ + libjpeg_turbo + libpng + torch.cxxdev + ] + ++ lib.optionals stdenv.hostPlatform.isDarwin [ + # This should match the SDK used by `torch` above + apple-sdk_13 + + # error: unknown type name 'MPSGraphCompilationDescriptor'; did you mean 'MPSGraphExecutionDescriptor'? + # https://developer.apple.com/documentation/metalperformanceshadersgraph/mpsgraphcompilationdescriptor/ + (darwinMinVersionHook "12.0") + ]; dependencies = [ numpy