pycuda>=2021.1
torch
torchvision
torchaudio