import sys if 'linux' in str(sys.platform): os.system('pip install torch==1.10.2+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html') else: os.system('pip install torch')