1. Скачайте cuda https://developer.nvidia.com/cuda-toolkit-archive
2. Установите куда
3. После завершения установки настройте переменные среды.
sudo vim ~/.bashrc
4. Добавьте в конце
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. Обратите внимание, что вместо указанного выше пути используется /usr/local /cuda-11.3, поскольку программные ссылки необходимы для обеспечения сосуществования нескольких версий CUDA. Введите следующий код, чтобы завершить создание программной ссылки, и /cuda-11.3замените его своим собственным именем каталога установки cuda.
а. Удалите созданную ранее мягкую ссылку
sudo rm -rf /usr/local/cuda
б. Создайте новую мягкую ссылку
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda
Если вам нужно изменить версию CUDA позже, просто выполните две приведенные выше команды еще раз, чтобы заменить программную ссылку.
6. Установка cuDNN скачать https://developer.nvidia.com/rdp/cudnn-archive
7. Загрузите и разархивируйте его.
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. Готово