Установить 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).