Coding on ubuntu and I install pytorch via pip with specific cuda version like:
--find-links https://download.pytorch.org/whl/torch_stable.html
torch==2.3.1+cu121
torchaudio==2.3.1+cu121
torchvision==0.18.1+cu121
And I noticed that many cuda dependencies were actually install automatically by pip such as nvidia-nvtx-cu12, nvidia-nvjitlink-cu12, nvidia-nccl-cu12, nvidia-curand-cu12, nvidia-cufft-cu12, nvidia-cuda-runtime-cu12, nvidia-cuda-nvrtc-cu12, nvidia-cuda-cupti-cu12, nvidia-cublas-cu12
Do I need to install cuda toolkit? Or those are already what I need to run pytorch using cuda.
1