Процесс установки Windows10+CUDA 10.1.0+CUDNN+pytorch

Основная цель написания этой статьи — записать и поделиться процессом успешной установки.Если у вас есть небольшой партнер, которого беспокоит эта проблема, вы также можете посмотреть, сможете ли вы найти решение здесь.

Без лишних слов, давайте сначала разберемся с моим общим процессом.

• Установить CUDA
• Установить cuDNN
• установить питорч
• контрольная работа

Первый шаг — установка CUDA

Все мы знаем, что установка CUDA — это использование графического процессора на видеокарте для ускорения операций, поэтому первое, что нужно определить, — поддерживает ли видеокарта вашего компьютера CUDA и установлен ли драйвер.

Сначала поговорим о драйвере, откройте его.Диспетчер устройств,оказатьсяАдаптер дисплея,

• Если вы видите модель вашей видеокарты, драйвер был успешно установлен.
• Если появляется желтый восклицательный знак, драйвер неисправен и его необходимо переустановить.

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

Драйвер, соответствующий GPU, версия CUDA

Вы можете проверить приведенную ниже таблицу, чтобы запросить версию графического процессора в соответствии с версией CUDA, которую вам нужно установить (не забудьте сначала проверить, в противном случае версия будет неверной, что приведет к успешной установке, но не будет работать). Проверьте URL-адрес соответствующая версия:docs.NVIDIA.com/bulky/bulky-he…

обновление драйвера

Как проверить, поддерживает ли ваша видеокарта CUDA

предоставить способ найтиПанель управления NVIDIAМой находится в строке состояния в правом нижнем углу. затем нажмитепомощь,компоненты,NVCUDA.DLLВы можете проверить версию CUDA, поддерживаемую видеокартой.

Скачать CUDA

Следующим шагом будет перейти на официальный сайт, чтобы скачать соответствующую версию CUDA.Я пошел прямо на официальный сайт и обнаружил, что прямая загрузка была версии 11, а мне нужна была версия 10.1.Позже через запрос я нашел тотАдреса разных версий CUDAЛюбая версия CUDA может быть загружена.

Затем выберите загрузку в соответствии с вашими потребностями.

Установить CUDA

После загрузки вы можете установить его в соответствии с шагами.Также есть предпосылка, что на вашем компьютере должна быть заранее установлена ​​версия vs2015 или выше. В процессе установки обратите внимание на два пути: первый — путь временной распаковки, а второй — путь установки. Если путь установки не указан по умолчанию, вам нужно его запомнить, потому что он будет использоваться при добавлении переменных среды позже.

Выше первый путь

выберитенастроить

Это второй путь, фактическое место установки, его рекомендуется не изменять, удобно добавить переменные окружения позже.

Проверьте, успешно ли установлена ​​CUDA

Открытымpowershell,илиcmdкомандное окно, введитеnvcc -VПроверьте, прошла ли установка успешно. На изображении выше показано, что установка прошла успешно.

Добавьте переменные среды для CUDA

Добавьте переменные среды. Компьютер -> Свойства -> Дополнительные параметры системы -> Переменные среды

Заполните в соответствии с вашим собственным путем установки:

CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1 
CUDA_LIB_PATH = %CUDA_PATH%\lib\x64 
CUDA_BIN_PATH = %CUDA_PATH%\bin 
CUDA_SDK_BIN_PATH = %CUDA_SDK_PATH%\bin\win64 
CUDA_SDK_LIB_PATH = %CUDA_SDK_PATH%\common\lib\x64

поставить вышеновыйв системные переменные

Затем найдите системную переменнуюpath,выберитередактировать.

Добавьте следующее:

%CUDA_LIB_PATH%;%CUDA_BIN_PATH%;%CUDA_SDK_LIB_PATH%;%CUDA_SDK_BIN_PATH%; C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64； C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64； C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64；

Второй шаг, установите cuDNN

иди первыйофициальный сайт CUDNNскачать. Зарегистрироваться в первый раз

выберитеCREATE AN ACCOUNT

Подтвердите свою электронную почту и заполните личную информацию.

Официальная страница загрузки CUDNN:developer.NVIDIA.com/Ren Cannon/Vinegar и др.-ааа…

Затем зарегистрируйте учетную запись и вернитесь к загрузке после регистрации.

Выберите версию, совместимую с CUDA.

Затем выберите cuDNN, подходящий для вашей системы.

После скачивания это установочный пакет, после распаковки будет三个文件夹. Скопируйте содержимое трех папок вПерейдите в соответствующую папку в вашей установке CUDA.. Например, если у вас есть библиотека после распаковки, скопируйте ее содержимое. Затем найдите путь к CUDA, также найдите папку lib и скопируйте содержимое в папку.

Третий шаг, установите pytorch

идти софициальный сайт питорчВыберите подходящую версию, затем скопируйте код в CMD или powershell.

Если перед установкой вы установили torch, pytorch и torchvision, удалите их.

pip uninstall torch

или использовать

conda uninstall torch

Я устанавливал его раньше, поэтому прямая установка не удалась, и я могу удалить его позже.

Последний шаг, тест

Откройте powershell, введите python,torch.codu.is_availableЕсли вывод True, это означает, что установка прошла успешно и ее можно использовать.

pytorch использует файлы whl для стабильной установки

1. Сначала найдите предыдущую версию в pytorch сейчас

2. Прокрутите вниз, чтобы найти веб-сайт, соответствующий версии (версия 1.8.0)

3. Скопируйте указанный выше путь в браузер, чтобы открыть

4. Для каждой версии pytorch есть файлы, выберите версию whl-файла, которую нужно загрузить.