b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

python3Packages.torchvision-bin: fix aarch64-darwin build

Browse source
This commit is contained in:
Samuel Ainsworth 2023-07-21 13:31:30 -07:00
parent 108644253e
commit 3080e7c051
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
pkgs/development/python-modules/torchvision/bin.nix
View file

@ -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 ];
}; };
} }