Version conflict on colab

[ShivanshuPurohit]

Describe the bug
Using pytorch 1.7.1 results in an error while keeping it at 1.7.0 (default on colab) rectifies it

To Reproduce
Steps to reproduce the behavior:
!pip install -r requirements.txt

Results in

Traceback (most recent call last):
  File "/usr/local/lib/python3.6/dist-packages/pkg_resources/__init__.py", line 568, in _build_master
    ws.require(__requires__)
  File "/usr/local/lib/python3.6/dist-packages/pkg_resources/__init__.py", line 886, in require
    needed = self.resolve(parse_requirements(requirements))
  File "/usr/local/lib/python3.6/dist-packages/pkg_resources/__init__.py", line 777, in resolve
    raise VersionConflict(dist, req).with_context(dependent_req)
pkg_resources.ContextualVersionConflict: (torch 1.7.1 (/usr/local/lib/python3.6/dist-packages), Requirement.parse('torch==1.7.0'), {'torchvision'})