Ubuntu устанавливает драйвер nvidia, cuda и cudnn

глубокое обучение

Это седьмой день моего участия в августовском испытании обновлений, подробности о мероприятии:Испытание августовского обновления

Программно-аппаратная среда

  • ubuntu 18.04 64bit
  • NVidia GTX 1070
  • CUDA 10.1
  • cudnn 7.6.3

установить драйвер нвидиа

Войти без установленного драйвераubuntuСистема, изображение на экране ненормальное, разрешение слишком маленькое. Скачать с официального сайтаRUNфайлы, такие как мойNVIDIA-Linux-x86_64-435.21.run, выбор версии здесь должен быть таким же, как показано ниже.cudaсоответствовать

# 添加执行权限
chmod a+x NVIDIA-Linux-x86_64-435.21.run
sudo ./NVIDIA-Linux-x86_64-435.21.run

Процесс установки в основном зависит от выбораyesилиno, перезагрузите систему после установки, вы обнаружите, что изображение на экране нормальное, а разрешение стало выше.

так какcudaСоответствующий драйвер уже включен в установочный файл, поэтому этот шаг фактически можно выполнить без предварительного, конечно, с этой установкой нет проблем, все еще зависит от личных привычек.

Установить CUDA

Далее запускаем установкуcuda, официальный адрес загрузки:developer.NVIDIA.com/rough-down…

# 添加可执行的权限
chmod +x cuda_10.1.243_418.87.00_linux.run
sudo ./cuda_10.1.243_418.87.00_linux.run

В процессе установки будет несколько вариантов, выберите в соответствии с вашей реальной ситуацией, например, использовать лиcudaВстроенный драйвер, путь установки, нужно ли устанавливатьsamplesи т.д. (разные для каждой версии)

ubuntu cuda

по умолчанию,cudaбыл установлен в/usr/local, а также создать мягкую ссылкуcuda

ubuntu cuda

Затем настройте переменные среды, отредактируйте~/.bashrcдокумент

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

Наконец выполнитьsource

source ~/.bashrc

Несколько важных переменных среды вступают в силу немедленно. Или снова открытьterminal, также вступает в силу.

установить cudnn

на официальный сайтdeveloper.nvidia.com/cudnnДля скачивания здесь требуется регистрация, что немного хлопотно. должны знать о том,cudnnверсия должна следоватьcudaсовпало

tar xvf cudnn-10.1-linux-x64-v7.6.3.36.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda-10.1/include
sudo cp -a cuda/lib64/libcudnn*  /usr/local/cuda-10.1/lib64

# 添加读的权限
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

контрольная работа

После завершения установки давайте проверим, соответствует ли средаok?

Входитьsampleв, сбежатьdeviceQuery

# 安装编译工具
sudo apt install build-essential

cd NVIDIA_CUDA-10.1_Samples/1_Utilities/deviceQuery
make

ubuntu cuda

окончательное исполнениеnvidia-smiВзгляни

ubuntu cuda

Я только что запустил пример здесь, чтобы вы могли видеть, что графический процессор в основном работает на полную мощность.

Сосуществование и переключение нескольких версий cuda

отcudaВы можете видеть, что установка нескольких версийcudaСосуществование возможно и переключение очень просто.

Процесс установки такой же, как было сказано выше, есть несколько версий для установки, например, я сказал, что здесь установлены 2, одна 10.0, другая 10.1, и версия 10.1 на данный момент активирована.

ubuntu cuda

Теперь давайте поработаем переключателем

# 删除软链接
sudo unlink /usr/local/cuda

# 重新创建软链接，以使用10.1为例
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

использованная литература

Категории