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

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

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

    В примере будет выбрано окружение, созданное с помощью CLIllama_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).

ML Space