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

评论

发表评论

了解 数据控|突破是我们的每一步 的更多信息

立即订阅以继续阅读并访问完整档案。

继续阅读