PyTorch CPU Package List
nvidia-cublas-cu12
nvidia-cuda-cupti-cu12
nvidia-cuda-nvrtc-cu12
nvidia-cuda-runtime-cu12
nvidia-cudnn-cu11
nvidia-cudnn-cu12
nvidia-cufft-cu12
nvidia-cufile-cu12
nvidia-curand-cu12
nvidia-cusolver-cu12
nvidia-cusparse-cu12
nvidia-cusparselt-cu12
nvidia-nccl-cu12
nvidia-nvjitlink-cu12
nvidia-nvtx-cu12
torch
torch-cuda80
torch-model-archiver
torch-tb-profiler
torchao
torchaudio
torchcodec
torchcsprng
torchdata
torchdistx
torchmetrics
torchrec
torchrec-cpu
torchserve
torchtext
torchtune
torchvision
triton