установка CUDA
скачать cuda
Введите команду nvidia-smi, чтобы просмотреть поддерживаемые версии cuda.
При наличии драйвера отображается следующая информация:
Если нет, отображается следующая информация:
Если его не видно, значит драйвер nvidia не установлен.Нажмите дополнительный драйвер и выберите драйвер соответствующей версии для его автоматической загрузки.
Нажмите «Применить изменения», затем установите драйвер, перезагрузите компьютер после завершения установки и проверьте, в порядке ли драйвер видеокарты.
Загрузите соответствующую версию cuda с https://developer.nvidia.com/cuda-toolkit-archive и выберите версию 18.04. Как показано ниже:
Следуйте инструкциям по установке cuda версии 10.1.
понижение версии gcc
Версия gcc, поставляемая с Ubuntu 20.04, — 9.7.0.Для установки cuda10.2 необходимо добавить gcc7.Введите команду для установки gcc7
sudo apt-get install gcc-7 g++-7
Глядя на версию gcc, можно увидеть, что на данный момент в системе есть две версии 7 и 9.
Чтобы использовать альтернативы обновлений для переключения версий, введите следующую команду:
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-9 50
На этом этапе введите команду sudo update-alternatives --config gcc, чтобы просмотреть версию gcc по умолчанию.Вы можете видеть, что текущая версия gcc по умолчанию — 7, то есть переключение прошло успешно.
установить cuda
Введите команду для установки
sudo sh cuda_10.2.89_440.33.01_linux.run
Нажмите «Продолжить».
введите принять
Установочный пакет cuda поставляется с собственным драйвером видеокарты, поэтому на этом этапе нажмите пробел, чтобы удалить опцию установки драйвера видеокарты, а затем выберите «Установить».
Настроить переменные среды
Введите команду gedit ~/.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}}
Обновить конфигурацию переменной среды
source ~/.bashrc
На этом установка cuda завершена, введите команду nvcc -V для просмотра информации cuda.
Установить cuDNN
Загрузите соответствующую версию cudnn с https://developer.nvidia.com/cudnn, вам необходимо авторизоваться.
После загрузки и распаковки скопируйте файл cuda/include/cudnn.h в папку usr/local/cuda/include, скопируйте все файлы из каталога cuda/lib64/ в папку /usr/local/cuda/lib64 и добавьте читать Получить разрешение:
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
На этом установка cuDNN завершена.