Чтобы поиграть с моделями ИИ, основой является вычислительная мощность графического процессора. Сначала необходимо понять информацию о CUDA и графическом процессоре вашего компьютера, установить соответствующее программное обеспечение и правильно его настроить.
Чтобы просмотреть информацию о CUDA и графическом процессоре вашего компьютера, вы можете просмотреть ее через командную строку.
Чтобы просмотреть информацию о графическом процессоре, откройте командную строку или PowerShell и введите следующую команду:
nvidia-smi
Эта команда отобразит модель графического процессора компьютера, версию драйвера и версию CUDA. Согласно информации, предоставленной nvidia-smi, используется графический процессор NVIDIA GeForce, а установленная на данный момент версия CUDA — 11.4.
Чтобы проверить версию CUDA Toolkit, откройте командную строку или PowerShell и введите следующую команду:
nvcc --version
Эта команда отобразит информацию о версии установленного набора инструментов CUDA.
Если nvcc не установлен, вам может потребоваться сначала установить набор инструментов CUDA. Чтобы установить набор инструментов CUDA в вашей системе, выполните следующие действия:
Загрузите установочный пакет CUDA Toolkit:
https://developer.nvidia.com/cuda-toolkit-archive Посетите страницу загрузки набора инструментов NVIDIA CUDA.
Выберите подходящую версию (например, CUDA 11.4) и вашу операционную систему.
Загрузите соответствующий установочный пакет.
Установите CUDA Toolkit: запустите загруженный установочный пакет и следуйте инструкциям для завершения установки.
После завершения установки вам необходимо добавить каталог CUDA bin и каталог libnvvp в переменные среды системы.
Откройте «Свойства системы». -> «Расширенные настройки системы» -> «Переменные среды».
В разделе «Системные переменные» найдите «Путь» и нажмите «Изменить».
Добавьте следующие пути (отрегулируйте в соответствии с вашим путем установки):
Скопировать код
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.4\libnvvp
Проверьте установку:
Откройте командную строку и введите следующую команду, чтобы проверить, успешно ли установлен набор инструментов CUDA:
nvcc --version
Вы должны увидеть информацию о версии компилятора CUDA.
Наконец, установите cuDNN
Посетите страницу загрузки NVIDIA cuDNN.
https://developer.nvidia.com/cudnn
Войдите в систему и загрузите файл cuDNN, соответствующий версии CUDA.
Затем установите cuDNN.