Установить PyTorch в текущее окружение
Рассмотрим пример создания окружения с PyTorch.
Создайте или выберите требуемое окружение.
В примере будет выбрано окружение, созданное с помощью CLI — llama_trainer.
conda activate -n llama_trainerУстановите 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).
См.также