pip install --upgrade torch torchvision
1. no module named "torch.cuda.amp"
2. ImportError: libgthread-2.0.so.0: cannot open shared object file: No such file or directory
apt-get update
apt-get install libglib2.0-dev
3. module 'distutils' has no attribute 'version'
pip install --upgrade torch
pip uninstall setuptools
pip install setuptools==58.0.4