1. Descarga cuda https://developer.nvidia.com/cuda-toolkit-archive
2. Instalar cuda
3. Una vez completada la instalación, configure las variables de entorno
sudo vim ~/.bashrc
4. Añadir al final
export LD_LIBRARY_PATH="/usr/local/cuda/lib64:$LD_LIBRARY_PATH"
export PATH="/usr/local/cuda/bin:$PATH"
export CUDA_HOME="/usr/local/cuda:$CUDA_HOME"
/cuda5. Tenga en cuenta que se usa /usr/local en lugar de la ruta anterior /cuda-11.3, porque se requieren enlaces suaves para lograr la coexistencia de varias versiones de CUDA. Ingrese el siguiente código para completar la generación del enlace suave y /cuda-11.3reemplácelo con su propio nombre de directorio de instalación de cuda.
a. Eliminar el enlace suave generado anteriormente
sudo rm -rf /usr/local/cuda
b. Generar un nuevo enlace suave
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda
Si necesita cambiar la versión de CUDA más tarde, simplemente ejecute los dos comandos anteriores nuevamente para reemplazar el enlace suave.
6. Descarga de instalación de cuDNN https://developer.nvidia.com/rdp/cudnn-archive
7. Descárgalo y descomprímelo
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn*.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
8. Terminado