Título del catálogo
1. Descarga el CUDA correspondiente
La premisa de que se ha instalado el controlador nvidia, verifique el Tutorial 1:
Verifique la correspondencia entre cuda y controlador (como se muestra en la figura): La cuda oficial y el controlador corresponden al
controlador que instalé es 440.100, por lo que debe seleccionarse CUDA10.2.89. Vaya a la descarga de CUDA , vaya al final de la página y seleccione descargar ahora, puede ver la versión antigua de cuda (como se muestra en la figura).
2. Instale CUDA
Seleccione CUDA Toolkit 10.2 en la guía de inicio rápido , consulte la página de descarga de la siguiente manera, ingrese el comando de solicitud en la terminal.
Tenga en cuenta que si la instalación falla, primero debe modificar los permisos del archivo de ejecución:
sudo chmod a+x cuda_10.2.89_440.33.01_linux.run
sudo sh cuda_10.2.89_440.33.01_linux.run
Luego comience a ingresar a la instalación, ingrese aceptar después de mucho texto para aceptar el acuerdo.
Indique que la instalación incompleta no importa aquí.
Configure las variables de entorno de cuda:
sudo su //切换为root超级用户的命令
sudo gedit ~/.bashrc //打开.bashrc文件
//在~/.bashrc的最后添加:
export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
Asegúrese de actualizar después de agregar: fuente ~ / .bashrc
3. Pruebe si CUDA se instaló correctamente
nvcc --version
Después del éxito, se muestra para
ejecutar un programa en las muestras para verificar:
cd /home/hy/NVIDIA_CUDA-10.2_Samples/1_Utilities/deviceQuery //输入自己的samples文件夹路径
sudo make
./deviceQuery
El resultado final result = PASS debe considerarse exitoso.
4. Instala cudnn
Primero, vaya al sitio web oficial para descargar cuDNN https://developer.nvidia.com/cudnn , debe registrarse como desarrollador para descargar. Seleccione el cudnn correspondiente a la versión cuda. El mío es cuda10.2, así que elija cudnn8, haga clic en la biblioteca cudnn para linux (x86) y descargue el archivo cudnn-xxxx.tgz.
descomprimir archivos:
tar xvzf cudnn-10.2-linux-x64-v8.0.1.13.tgz
Una vez completada la descompresión, obtendrá una carpeta cuda, que tiene dos carpetas que incluyen include y lib64. Cambie a la ruta cuda / include y realice las siguientes operaciones:
sudo cp cudnn.h /usr/local/cuda/include/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
Cambiar a cuda / lib64:
sudo cp lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
5. Verifique la versión de cudnn
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 8
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 1
-
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include “driver_types”