cuda

install CUDA

• https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local
• https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html#configuration
• https://docs.nvidia.com/cuda/wsl-user-guide/index.html

install pytorch with CUDA

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

test

import torch

if torch.cuda.is_available():
    print("CUDA is beschikbaar