Enables use PT nightlies on CPU

Author: agunapal

ts_scripts/install_dependencies.py

@@ -124,8 +124,12 @@ def install_python_packages(self, cuda_version, requirements_file_path, nightly)
 
         # Install PyTorch packages
         if nightly:
+            if cuda_version:
+                pt_nightly = cuda_version
+            else:
+                pt_nightly = "cpu"
             os.system(
-                f"pip3 install numpy --pre torch torchvision torchtext torchaudio --force-reinstall --extra-index-url https://download.pytorch.org/whl/nightly/{cuda_version}"
+                f"pip3 install numpy --pre torch torchvision torchtext torchaudio --force-reinstall --extra-index-url https://download.pytorch.org/whl/nightly/{pt_nightly}"
             )
         else:
             self.install_torch_packages(cuda_version)