torch  # pytorch cudatoolkit
torchvision