torch: Only remove FindCUDAToolkit.cmake if it exists

datakami · Jul 24, 2024 · f08a1e3 · f08a1e3
1 parent 177ea1e
commit f08a1e3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/overrides.nix b/overrides.nix
@@ -7,7 +7,7 @@
 
     # https://github.com/NixOS/nixpkgs/blob/master/pkgs/development/python-modules/torch/fix-cmake-cuda-toolkit.patch
     mkDerivation.postInstall = ''
-      rm $out/${config.deps.python.sitePackages}/torch/share/cmake/Caffe2/FindCUDAToolkit.cmake
+      rm -f $out/${config.deps.python.sitePackages}/torch/share/cmake/Caffe2/FindCUDAToolkit.cmake
     '';
     # seems to be a bug in the torch-cu117 package
     env.autoPatchelfIgnoreMissingDeps = [ "libnvToolsExt.so.1" ];