python3 安装pytorch

• 官网 https://pytorch.org/

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')