George V. Reilly georgevreilly

CUDA deps cannot be preloaded under Bazel

🐛 Describe the bug

If Torch 2.1.0 is used as a dependency with Bazel and rules_python, _preload_cuda_deps fails with OSError: libcufft.so.11: cannot open shared object file: No such file or directory.

Torch 2.1 works fine if you install it and its CUDA dependencies into a single site-packages (e.g., in a virtualenv). It doesn't work with Bazel, as it installs each dependency into its own directory tree, which is appended to PYTHONPATH.

$ bazel test //...

Words with all five vowels in forward and backward order

Using /usr/share/dict/words on macOS Sonoma 14.2.1

Forward order, aeiou

180 results

$ grep -i '.*a.*e.*i.*o.*u.*' /usr/share/dict/words

	#!/usr/bin/env python3

	from __future__ import annotations

	import argparse
	import random
	from functools import reduce

	from typing import Any

	#!/usr/bin/env python3

	"""
	Produce topological sorted graphs of shared object library dependencies.
	"""

	from __future__ import annotations

	import json
	import os