Upgrade CUDA version used in PyTorch cuda12-variant to 12.6 (#2291)

Author: mathbunnyru

images/pytorch-notebook/cuda12/Dockerfile

@@ -13,7 +13,7 @@ SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
 # Install PyTorch with pip (https://pytorch.org/get-started/locally/)
 # hadolint ignore=DL3013
-RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu124' \
+RUN pip install --no-cache-dir --extra-index-url=https://pypi.nvidia.com --index-url 'https://download.pytorch.org/whl/cu126' \
     'torch' \
     'torchaudio' \
     'torchvision' && \