Using code in the new Gpu-s(Python Versions)

[ghost]

Hi ! I am trying to use this code in Google Colab and the latest problem that I am facing is the issue that the code is compiled in old cuda version. The error that I am getting is verbatim as follows :

"Found GPU0 Tesla T4 which requires CUDA_VERSION >= 9000 for
optimal performance and fast startup time, but your PyTorch was compiled
with CUDA_VERSION 8000. Please install the correct PyTorch binary
using instructions from http://pytorch.org"
Is there any structured way how to compile the code ? Can I try to change the code so it can work in the new Pytorch versions ? Will the current pretrained stuff work ? Please any advice or suggestion would be helpful ! Or at least if you could tell me how you have been able to run the code ? Thanks in advance!