Commits on Dec 7, 2022

1. Add manual cuda deps search logic (pytorch#90411) …

   If PyTorch is package into a wheel with [nvidia-cublas-cu11](https://pypi.org/project/nvidia-cublas-cu11/), which is designated as PureLib, but `torch` wheel is not, can cause a torch_globals loading problem.
   
   Fix that by searching for `nvidia/cublas/lib/libcublas.so.11` an `nvidia/cudnn/lib/libcudnn.so.8` across all `sys.path` folders.
   
   Test plan:
   ```
   docker pull amazonlinux:2
   docker run --rm -t amazonlinux:2 bash -c 'yum install -y python3 python3-devel python3-distutils patch;python3 -m pip install torch==1.13.0;curl -OL https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/90411.diff; pushd /usr/local/lib64/python3.7/site-packages; patch -p1 </90411.diff; popd; python3 -c "import torch;print(torch.__version__, torch.cuda.is_available())"'
   ```
   
   Fixes pytorch#88869
   
   Pull Request resolved: pytorch#90411
   Approved by: https://github.com/atalman

   

   malfet authored and atalman committed Dec 7, 2022
   Copy the full SHA

   cff9766 View commit details

   Browse the repository at this point in the history