install a version of PyTorch compatible with the built-in CUDA drivers

Create requirements.txt