# Python extras for the ResNet-18 / CIFAR-10 CUDA smoke test.
# Install the CUDA wheel on the GPU box, not the CPU wheel:
#   pip install -r examples/python/requirements.txt --index-url https://download.pytorch.org/whl/cu124
torch>=2.1
torchvision>=0.16
