pytorch报错RuntimeError: DataLoader worker (pid 18906) is killed by signal: Segmentation fault.

系统环境：
ubuntu16.04
cuda11

torch包的版本
torch 1.8.1
torchaudio 0.8.1
torchvision 0.9.1

运行官方的tv-training-code.py时出现错误
RuntimeError: DataLoader worker (pid 18906) is killed by signal: Segmentation fault.

找了半天应该是dataloader问题，解决办法：
方法1：
num_workers设为0
即

torch.utils.data.DataLoader(
        dataset_test, batch_size=2, shuffle=False, num_workers=0,
        collate_fn=utils.collate_fn)

方法2
使用torch1.9，即可解决

conda create -n condapy python=3.7
pip  install torch torchvision torchaudio