pip install torch==2.1.0
pip install torchvision==0.16.0
pip install torchaudio==2.1.0