Установить 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).
См.также
Была ли статья полезной ?
Предыдущая статья
Отправить и получить окружение
Следующая статья
Обновить окружение с помощью yml-файла