Установка PyTorch в текущее окружение

Рассмотрим пример создания окружения с PyTorch.

  1. Создайте или выберите требуемое окружение.

    В примере будет выбрано окружение, созданное с помощью CLI — llama_trainer.

    conda activate -n llama_trainer

  2. Установите PyTorch:

    pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121

При выборе версии PyTorch учитывайте версию CUDA. Для этого изучите особенности, описанные в документации PyTorch (en).

См.также

Описание команд и параметров

Запустили Evolution free tier
для Dev & Test
Получить