object_detection / requirements_gpu.txt
Sudipta Nayak
torch with CPU support only
2da0893
raw
history blame
196 Bytes
# pytorch with CUDA support
-i https://download.pytorch.org/whl/cu113
# CUDA 11.3
# torch==1.11.0+cu113
# torchvision==0.12.0+cu113
#CPU-only version of PyTorch
torch==1.11.0
torchvision==0.12.0