numpy==1.23 --extra-index-url https://download.pytorch.org/whl/cu118 torch==2.0.0+cu118 torchvision==0.15.0+cu118 torchaudio==2.0.0+cu118