-
-
Save cleverca22/0d7180681a76b6e12a99e86b309c0e52 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
diff --git a/pkgs/development/python-modules/pytorch/default.nix b/pkgs/development/python-modules/pytorch/default.nix | |
index 0f6cb16bf75..fb6e3d6acdc 100644 | |
--- a/pkgs/development/python-modules/pytorch/default.nix | |
+++ b/pkgs/development/python-modules/pytorch/default.nix | |
@@ -134,6 +134,7 @@ in buildPythonPackage rec { | |
'' + lib.optionalString (cudaSupport && cudnn != null) '' | |
export CUDNN_INCLUDE_DIR=${cudnn}/include | |
''; | |
+ postConfigure = "exit 1"; | |
preFixup = '' | |
function join_by { local IFS="$1"; shift; echo "$*"; } | |
@@ -203,6 +204,7 @@ in buildPythonPackage rec { | |
cp -r $out/${python.sitePackages}/torch/lib $dev/lib | |
cp -r $out/${python.sitePackages}/torch/include $dev/include | |
cp -r ${src}/c10/* $dev/include/c10/ | |
+ find -name cuda_cmake_macros.h | |
''; | |
postFixup = stdenv.lib.optionalString stdenv.isDarwin '' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
let | |
pkgs = import ./. { config.allowUnfree = true; }; | |
in pkgs.stdenv.mkDerivation { | |
name = "c10"; | |
src = pkgs.python37Packages.pytorch.src; | |
postUnpack = '' | |
sourceRoot+=/c10/cuda | |
echo source root reset to $sourceRoot | |
''; | |
nativeBuildInputs = [ pkgs.cmake ]; | |
buildInputs = [ pkgs.cudatoolkit ]; | |
} |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment