Как установить PyTorch на Windows 10 / 11 [Nvidia AMD GPU и CPU]

1. Предварительные требования для установки PyTorch
2. Доступ к странице установки PyTorch
3. Выбор правильной вычислительной платформы
4. Проверка версии CUDA
5. Скачивание CUDA Toolkit
6. Установка CUDA
7. Установка Visual Studio
8. Завершение установки CUDA
9. Установка PyTorch
10. Проверка установки PyTorch
11. Часто задаваемые вопросы

Предварительные требования для установки PyTorch

Перед установкой PyTorch убедитесь, что Python и pip установлены на вашем компьютере. Если вы еще не установили их, обратитесь к надежным онлайн-ресурсам за помощью. После того как вы настроите Python и pip, вы можете приступить к установке PyTorch.

Доступ к странице установки PyTorch

Чтобы начать процесс установки, перейдите на официальный сайт PyTorch. На главной странице прокрутите вниз, чтобы найти варианты установки. Выберите стабильную версию PyTorch, которая в настоящее время является версией 2.1.1. Выберите вашу операционную систему; для пользователей Windows выберите Windows. Если у вас нет Conda, оставьте опцию пакета установленной на pip и подтвердите, что Python выбран в качестве языка.

Выбор правильной вычислительной платформы

Опция вычислительной платформы относится к типу графической карты, которой вы обладаете. Если у вас есть графическая карта Nvidia, выберите соответствующую версию CUDA. Для графических карт AMD выберите ROCm. Если в вашей системе нет GPU, выберите опцию CPU. Обратите внимание, что опция графической карты AMD недоступна для пользователей Windows; она требует операционной системы Linux.

Проверка версии CUDA

Перед выполнением команды pip install важно проверить установленную версию CUDA. Откройте PowerShell и введите команду 'nvcc --version', чтобы проверить вашу установку CUDA. Если CUDA не установлена, вам нужно будет скачать ее с официального сайта Nvidia.

Скачивание CUDA Toolkit

Чтобы установить CUDA, поищите в вашем браузере 'install CUDA 12.1' и нажмите на первую ссылку с сайта Nvidia. Выберите вашу операционную систему, архитектуру (x86_64) и версию Windows. Выберите исполняемый локальный установщик и убедитесь, что у вас есть как минимум 3.1 ГБ свободного места перед началом загрузки.

Установка CUDA

После того как CUDA Toolkit будет загружен, перейдите в папку загрузок и дважды щелкните на исполняемом файле. Следуйте инструкциям, чтобы извлечь файлы и согласиться с лицензионным соглашением. Если вам будет предложено установить совместимость с Visual Studio, вам нужно будет установить Visual Studio для полной функциональности.

Установка Visual Studio

Чтобы установить Visual Studio, найдите ее в интернете и перейдите на страницу загрузок. Скачайте бесплатную версию Community Edition. После загрузки откройте установщик и выберите пакет разработки Python, а также любые другие пакеты, которые могут вам понадобиться. Нажмите кнопку установки, чтобы продолжить.

Завершение установки CUDA

После установки Visual Studio вернитесь к процессу установки CUDA. Снова откройте исполняемый файл CUDA и следуйте инструкциям, чтобы завершить установку. После завершения проверьте установку, выполнив 'nvcc --version' в PowerShell, чтобы подтвердить, что CUDA 12.1 теперь установлена.

Установка PyTorch

С успешно установленной CUDA вы можете теперь приступить к установке PyTorch. Откройте PowerShell и выполните команду pip install для PyTorch. Это загрузит и установит необходимые пакеты для PyTorch на вашей операционной системе Windows 10 или 11.

Проверка установки PyTorch

Чтобы убедиться, что PyTorch установлен правильно, выполните несколько команд в PowerShell. Начните с ввода 'python', чтобы получить доступ к интерпретатору Python. Затем импортируйте PyTorch, введя 'import torch'. Чтобы проверить, доступна ли CUDA, используйте команду 'torch.cuda.is_available()'. Если она возвращает true, ваша установка прошла успешно.

Часто задаваемые вопросы

В: Каковы предварительные требования для установки PyTorch?
О: Перед установкой PyTorch убедитесь, что Python и pip установлены на вашем компьютере.
В: Как мне получить доступ к странице установки PyTorch?
О: Перейдите на официальный сайт PyTorch и прокрутите вниз, чтобы найти варианты установки.
В: Как мне выбрать правильную вычислительную платформу для PyTorch?
О: Выберите соответствующую версию CUDA для графических карт Nvidia, ROCm для AMD или выберите опцию CPU, если у вас нет GPU.
В: Как я могу проверить свою версию CUDA?
О: Откройте PowerShell и введите команду 'nvcc --version', чтобы проверить вашу установку CUDA.
В: Как мне скачать CUDA Toolkit?
О: Поиските 'install CUDA 12.1' в вашем браузере и выберите соответствующие параметры на сайте Nvidia.
В: Какие шаги включены в установку CUDA?
О: Дважды щелкните на загруженном исполняемом файле CUDA, следуйте инструкциям и согласитесь с лицензионным соглашением.
В: Как мне установить Visual Studio?
О: Скачайте версию Community Edition с сайта Visual Studio и выберите пакет разработки Python во время установки.
В: Что мне делать после установки Visual Studio?
О: Вернитесь к процессу установки CUDA и завершите установку, следуя инструкциям.
В: Как мне установить PyTorch?
О: Откройте PowerShell и выполните команду pip install для PyTorch после успешной установки CUDA.
В: Как я могу проверить, что PyTorch установлен правильно?
О: Выполните 'python' в PowerShell, затем 'import torch' и проверьте с помощью 'torch.cuda.is_available()'.