Enables using PyTorch nightlies on CPU

[agunapal]

Description

Fixes the build script to use PT nightlies on CPU

Fixes #(issue)

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

Feature/Issue validation/testing

• CPU

python ./ts_scripts/install_dependencies.py --nightly_torch
Collecting numpy                                                                                                                                                                          
  Downloading numpy-1.26.4-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl.metadata (61 kB)                                                                                    
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 61.0/61.0 kB 2.5 MB/s eta 0:00:00
Collecting torch                                                                                                                                                                          
  Downloading https://download.pytorch.org/whl/nightly/cpu/torch-2.3.0.dev20240222%2Bcpu-cp310-cp310-linux_x86_64.whl (188.9 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 188.9/188.9 MB 12.7 MB/s eta 0:00:00
Collecting torchvision
  Downloading https://download.pytorch.org/whl/nightly/cpu/torchvision-0.18.0.dev20240222%2Bcpu-cp310-cp310-linux_x86_64.whl (1.7 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 1.7/1.7 MB 59.1 MB/s eta 0:00:00

pip list | grep -i torch
torch                     2.3.0.dev20240222+cpu
torchaudio                2.2.0.dev20240222+cpu
torchdata                 0.7.1.dev20240221+cpu
torchtext                 0.17.0.dev20240222+cpu
torchvision               0.18.0.dev20240222+cpu

• CUDA

python ./ts_scripts/install_dependencies.py --cuda=cu121 --nightly_torch

pip list | grep -i torch
pytorch-triton            3.0.0+901819d2b6
torch                     2.3.0.dev20240222+cu121
torchaudio                2.2.0.dev20240222+cu121
torchtext                 0.6.0
torchvision               0.18.0.dev20240222+cu121

Checklist:

• Did you have fun?
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

[mreso]

LGTM