Actualice CUDNN a 8.2 en google colab.
Quiero actualizar la versión de CUDNN de 8.0 a 8.1 y la versión de CUDA a 11.2, pero no estoy seguro de cómo hacerlo en Colab. A continuación, se muestra el script que escribí para ejecutar en Google Colab, pero no cambia nada. ¿Qué cambios debo hacer aquí? ¿O hay alguna otra manera de hacerlo?
!sudo apt install cuda-toolkit-11-2
!wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
!sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
!sudo apt-key adv –fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
!sudo add-apt-repository “deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /”
!sudo apt-get update
!sudo apt-get -y install cuda
!sudo apt –fix-broken install
!sudo apt install cuda-cudart-11-2
import os
os.environ[‘PATH’] = ‘/usr/local/cuda-11.2/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/tools/node/bin:/tools/google-cloud-sdk/bin:/opt/bin’
os.environ[‘LD_LIBRARY_PATH’] = ‘/usr/local/cuda-11.2/lib64:/usr/local/cuda-11.2/lib64:/usr/lib64-nvidia’
!cp “/content/drive/My Drive/cudnn-11.2-linux-x64-v8.1.0.0.tgz” “/content/cudnn-11.2.tgz”
!mkdir cudnn
!tar -C cudnn/ -zxvf cudnn-11.2.tgz
%cd cudnn/
!sudo cp cuda/include/cudnn.h /usr/local/cuda-11.2/include
!sudo cp cuda/lib64/libcudnn /usr/local/cuda-11.2/lib64
!sudo chmod a+r /usr/local/cuda-11.2/include/cudnn.h /usr/local/cuda-11.2/lib64/libcudnn
!sudo ldconfig
%cd /content/
Comprobar la versión de CUDA y CUDNN
! nvcc –version
!nvidia-smi
import chainer
print(“—————–INFO DEL SISTEMA———————-“)
chainer.print_runtime_info()
Estoy tratando de utilizar la detección de objetos de TensorFlow y siguiendo este enlace para comprobar las versiones compatibles de CUDA y CUDNN: enlace.
davy.ai
_soporte
Parece que estás intentando actualizar la versión de CUDNN de 8.0 a 8.1 y la versión de CUDA a 11.2. Para ello, debes asegurarte de que está instalado en tu sistema el Kit de herramientas CUDA 11.2. Puedes hacerlo ejecutando el siguiente comando:
sudo apt install cuda-toolkit-11-2También necesitas descargar e instalar la biblioteca CUDNN 8.1. Puedes hacerlo ejecutando los siguientes comandos:
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-ubuntu1604.pin
sudo mv cuda-ubuntu1604.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo add-apt-repository "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda
sudo apt --fix-broken install
sudo apt install cuda-cudart-11-2
Una vez instalado el Kit de herramientas CUDA 11.2, puedes descargar e instalar la biblioteca CUDNN 8.1 ejecutando los siguientes comandos:
cp "/content/drive/My Drive/cudnn-11.2-linux-x64-v8.1.0.tgz" "/content/cudnn-11.2.tgz"
mkdir cudnn
tar -C cudnn/ -zxvf cudnn-11.2.tgz
cd cudnn/
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.2/include
sudo cp cuda/lib64/libcudnn /usr/local/cuda-11.2/lib64
sudo chmod a+r /usr/local/cuda-11.2/include/cudnn.h /usr/local/cuda-11.2/lib64/libcudnn
sudo ldconfig
cd /content/
Una vez instalada la biblioteca CUDNN 8.1, puedes comprobar las versiones de CUDA y CUDNN ejecutando los siguientes comandos:
nvcc --version
nvidia-smi
import chainer
print("-----------------INFORMACIÓN DEL SISTEMA----------------------")
chainer.print_runtime_info()
Si las versiones de CUDA y CUDNN están correctamente instaladas, deberías poder usar la detección de objetos de Tensorflow.