`test_models.py` segfaults on `r2plus1d_18_cuda` on the `unittest_linux_gpu_py3.8 ` CI job. See e.g. failures in https://github.com/pytorch/vision/pull/3700 or https://github.com/pytorch/vision/pull/3699 cc @seemethere