Just delete version specification.
-m pip install torch==2.1.2 torchvision --extra-index-url https://download.pytorch.org/whl/cu121
Just delete version specification.
-m pip install torch==2.1.2 torchvision --extra-index-url https://download.pytorch.org/whl/cu121