Why torchtext needs to reinstall torch

[WhenMelancholy]

Hi team, I am trying to install torchtext with torch 2.2.1-cu121 installed. But once I run pip install torchtext the pip will install torch 2.2.1 cpu version for me, is there any way to avoid this?

The output log:

Successfully installed torch-2.2.2+cu121 torchaudio-2.2.2+cu121 torchvision-0.17.2+cu121
PS :/scratch/github/scgpt$ pip uninstall torchtext
Found existing installation: torchtext 0.17.1
Uninstalling torchtext-0.17.1:
  Would remove:
    /anaconda/envs/scgpt/lib/python3.11/site-packages/torchtext-0.17.1.dist-info/*
    /anaconda/envs/scgpt/lib/python3.11/site-packages/torchtext/*
Proceed (Y/n)? 
  Successfully uninstalled torchtext-0.17.1
PS :/scratch/github/scgpt$ pip install -U torchtext --no-cache
Collecting torchtext
  Downloading torchtext-0.17.1-cp311-cp311-manylinux1_x86_64.whl.metadata (7.6 kB)
Requirement already satisfied: tqdm in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torchtext) (4.66.2)
Requirement already satisfied: requests in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torchtext) (2.28.1)
Collecting torch==2.2.1 (from torchtext)
  Downloading torch-2.2.1-cp311-cp311-manylinux1_x86_64.whl.metadata (26 kB)
Requirement already satisfied: numpy in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torchtext) (1.26.3)
Requirement already satisfied: torchdata==0.7.1 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torchtext) (0.7.1)
Requirement already satisfied: filelock in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (3.9.0)
Requirement already satisfied: typing-extensions>=4.8.0 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (4.8.0)
Requirement already satisfied: sympy in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (1.12)
Requirement already satisfied: networkx in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (3.2.1)
Requirement already satisfied: jinja2 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (3.1.2)
Requirement already satisfied: fsspec in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (2024.2.0)
Requirement already satisfied: nvidia-cuda-nvrtc-cu12==12.1.105 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.105)
Requirement already satisfied: nvidia-cuda-runtime-cu12==12.1.105 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.105)
Requirement already satisfied: nvidia-cuda-cupti-cu12==12.1.105 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.105)
Requirement already satisfied: nvidia-cudnn-cu12==8.9.2.26 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (8.9.2.26)
Requirement already satisfied: nvidia-cublas-cu12==12.1.3.1 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.3.1)
Requirement already satisfied: nvidia-cufft-cu12==11.0.2.54 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (11.0.2.54)
Requirement already satisfied: nvidia-curand-cu12==10.3.2.106 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (10.3.2.106)
Requirement already satisfied: nvidia-cusolver-cu12==11.4.5.107 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (11.4.5.107)
Requirement already satisfied: nvidia-cusparse-cu12==12.1.0.106 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.0.106)
Requirement already satisfied: nvidia-nccl-cu12==2.19.3 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (2.19.3)
Requirement already satisfied: nvidia-nvtx-cu12==12.1.105 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (12.1.105)
Requirement already satisfied: triton==2.2.0 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torch==2.2.1->torchtext) (2.2.0)
Requirement already satisfied: urllib3>=1.25 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from torchdata==0.7.1->torchtext) (1.26.13)
Requirement already satisfied: nvidia-nvjitlink-cu12 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from nvidia-cusolver-cu12==11.4.5.107->torch==2.2.1->torchtext) (12.4.99)
Requirement already satisfied: charset-normalizer<3,>=2 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from requests->torchtext) (2.1.1)
Requirement already satisfied: idna<4,>=2.5 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from requests->torchtext) (3.4)
Requirement already satisfied: certifi>=2017.4.17 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from requests->torchtext) (2022.12.7)
Requirement already satisfied: MarkupSafe>=2.0 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from jinja2->torch==2.2.1->torchtext) (2.1.3)
Requirement already satisfied: mpmath>=0.19 in /anaconda/envs/scgpt/lib/python3.11/site-packages (from sympy->torch==2.2.1->torchtext) (1.3.0)
Downloading torchtext-0.17.1-cp311-cp311-manylinux1_x86_64.whl (2.0 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 2.0/2.0 MB 45.3 MB/s eta 0:00:00
Downloading torch-2.2.1-cp311-cp311-manylinux1_x86_64.whl (755.6 MB)
   ━━━━━━━━━━━━━━━━━━━━━━━━━━━╸━━━━━━━━━━━━ 526.0/755.6 MB 289.7 MB/s eta 0:00:01
ERROR: Operation cancelled by user