Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
Nix
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465{ lib, stdenv, cmake, libtorch-bin, linkFarm, symlinkJoin,
cudaSupport, cudaPackages ? { },}:let inherit (cudaPackages) cudatoolkit cudnn;
cudatoolkit_joined = symlinkJoin { name = "${cudatoolkit.name}-unsplit"; paths = [ cudatoolkit.out cudatoolkit.lib ]; };
# We do not have access to /run/opengl-driver/lib in the sandbox, # so use a stub instead. cudaStub = linkFarm "cuda-stub" [ { name = "libcuda.so.1"; path = "${cudatoolkit}/lib/stubs/libcuda.so"; } ];
instdenv.mkDerivation { pname = "libtorch-test"; version = libtorch-bin.version;
src = lib.fileset.toSource { root = ./.; fileset = lib.fileset.unions [ ./CMakeLists.txt ./test.cpp ]; };
nativeBuildInputs = [ cmake ];
buildInputs = [ libtorch-bin ] ++ lib.optionals cudaSupport [ cudnn ];
cmakeFlags = lib.optionals cudaSupport [ "-DCUDA_TOOLKIT_ROOT_DIR=${cudatoolkit_joined}" ];
doCheck = true;
installPhase = '' touch $out '';
checkPhase = lib.optionalString cudaSupport '' LD_LIBRARY_PATH=${cudaStub}''${LD_LIBRARY_PATH:+:}$LD_LIBRARY_PATH \ '' + '' ./test '';}