PyTorch update

Change PyTorch version to 1.9.0 in library requirements and Python to 3.8.