python3Packages.ale-py: add missing nvcc to fix CUDA build
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
{
|
||||
lib,
|
||||
stdenv,
|
||||
config,
|
||||
buildPythonPackage,
|
||||
fetchFromGitHub,
|
||||
|
||||
@@ -17,6 +18,9 @@
|
||||
opencv,
|
||||
zlib,
|
||||
|
||||
# nativeBuildInputs
|
||||
cudaPackages,
|
||||
|
||||
# dependencies
|
||||
numpy,
|
||||
|
||||
@@ -49,6 +53,11 @@ buildPythonPackage (finalAttrs: {
|
||||
jax
|
||||
];
|
||||
|
||||
nativeBuildInputs = lib.optionals config.cudaSupport [
|
||||
# Required by opencv's cmake
|
||||
cudaPackages.cuda_nvcc
|
||||
];
|
||||
|
||||
# disable lto on darwin, cmake cannot find llvm-ar
|
||||
postPatch = lib.optionalString stdenv.hostPlatform.isDarwin ''
|
||||
substituteInPlace src/ale/CMakeLists.txt \
|
||||
|
||||
Reference in New Issue
Block a user