VSCode image with PyTorch and GPU persistence