audio: Torchaudio + tensorflow + CUDA 11.0 = segfault

🐛 Bug

Importing torchaudio after tensorflow-gpu while using CUDA 11.0 causes a segfault. This issue was originally reported in the AugLy repo: https://github.com/facebookresearch/AugLy/issues/28.

To Reproduce

Steps to reproduce the behavior:

Only happens on CUDA 11.0, so we haven’t been able to reproduce this error.

import tensorflow
import augly.audio as audaugs

Output:

2021-06-18 18:42:04.241048: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
Segmentation fault (core dumped)

Expected behavior

No segfault.

Environment

• What commands did you used to install torchaudio (conda/pip/build from source)?

• pip install -r torch==1.8.1 torchaudio==0.8.1 (in AugLy’s requirements.txt)

• PyTorch Version (e.g., 1.0):

• OS (e.g., Linux):

• How you installed PyTorch (conda, pip, source):

• Build command you used (if compiling from source):

• Python version:

• CUDA/cuDNN version: 11.0

• GPU models and configuration:

• Any other relevant information:

Additional context

This problem doesn’t happen on our (AugLy’s maintainers’) environments, only on one user’s.