Instale nccl (libra de comunicaciones colectivas de Nvidia) para CUDA 12.4. Bsicamente, es NCCL 2.20.5, que se lanz el 5 de marzo de 2024. Puede encontrarlo en el sitio web de Nvidia de la siguiente manera: -download . Ejecute los comandos para la instalacin de red. A continuacin, necesitars instalar Nvidia cuDNN. Incluso si crees que lo tienes, vuelve a realizar los pasos. Puede ir a la pgina de descarga de cuDNN de Nvidia para obtener instrucciones. Finalmente, el ltimo paso pero el ms importante es reinstalar PyTorch. Excepto usar la compilacin para obtener la ltima versin: