nvidia-smi
Here you can see that the CUDA version is 12.6.
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
Check the local CUDA version and install torch
nvidia-smi
Here you can see that the CUDA version is 12.6.
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
发表评论