python3Packages.torchvision-bin: fix aarch64-darwin build
This commit is contained in:
parent
108644253e
commit
3080e7c051
|
@ -28,13 +28,14 @@ in buildPythonPackage {
|
||||||
|
|
||||||
disabled = (pythonOlder "3.8") || (pythonAtLeast "3.12");
|
disabled = (pythonOlder "3.8") || (pythonAtLeast "3.12");
|
||||||
|
|
||||||
buildInputs = with cudaPackages; [
|
# Note that we don't rely on config.cudaSupport here, because the Linux wheels all come built with CUDA support.
|
||||||
|
buildInputs = with cudaPackages; lib.optionals stdenv.isLinux [
|
||||||
# $out/${sitePackages}/torchvision/_C.so wants libcudart.so.11.0 but torchvision.libs only ships
|
# $out/${sitePackages}/torchvision/_C.so wants libcudart.so.11.0 but torchvision.libs only ships
|
||||||
# libcudart.$hash.so.11.0
|
# libcudart.$hash.so.11.0
|
||||||
cuda_cudart
|
cuda_cudart
|
||||||
];
|
];
|
||||||
|
|
||||||
nativeBuildInputs = [
|
nativeBuildInputs = lib.optionals stdenv.isLinux [
|
||||||
autoPatchelfHook
|
autoPatchelfHook
|
||||||
addOpenGLRunpath
|
addOpenGLRunpath
|
||||||
];
|
];
|
||||||
|
@ -49,7 +50,7 @@ in buildPythonPackage {
|
||||||
|
|
||||||
pythonImportsCheck = [ "torchvision" ];
|
pythonImportsCheck = [ "torchvision" ];
|
||||||
|
|
||||||
preInstall = ''
|
preInstall = lib.optionalString stdenv.isLinux ''
|
||||||
addAutoPatchelfSearchPath "${torch-bin}/${python.sitePackages}/torch"
|
addAutoPatchelfSearchPath "${torch-bin}/${python.sitePackages}/torch"
|
||||||
'';
|
'';
|
||||||
|
|
||||||
|
@ -62,7 +63,7 @@ in buildPythonPackage {
|
||||||
# https://www.intel.com/content/www/us/en/developer/articles/license/onemkl-license-faq.html
|
# https://www.intel.com/content/www/us/en/developer/articles/license/onemkl-license-faq.html
|
||||||
license = licenses.bsd3;
|
license = licenses.bsd3;
|
||||||
sourceProvenance = with sourceTypes; [ binaryNativeCode ];
|
sourceProvenance = with sourceTypes; [ binaryNativeCode ];
|
||||||
platforms = [ "x86_64-linux" ];
|
platforms = [ "aarch64-darwin" "x86_64-darwin" "x86_64-linux" ];
|
||||||
maintainers = with maintainers; [ junjihashimoto ];
|
maintainers = with maintainers; [ junjihashimoto ];
|
||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue