Предварительно скомпилированные версии Pytorch

Я только что обнаружил о репозитории, который только что спас мой бекон, поэтому я решил поделиться им с вами, ребята.

Если вы перейдете на веб -сайт Pytorch, вам будет предоставлено много вариантов его установки:

Дело в том, что если вы хотите установить другую версию, чем указанная (1.7.1 на сегодняшний день), вы начнете не очень приятную охоту за сокровищами.

Есть репо, который как -то понадобился мне довольно много времени, чтобы найти здесь:

https://download.pytorch.org/whl/torch_stable.html

Но вам не нужно нажимать на это! Вы можете спросить Пип Чтобы получить это для вас:

pip[3] install pytorch==1.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

И Voilá, ваша посылка будет установлена автоматически!

Теперь некоторые дополнительные советы:

1. Некоторые CUDA Версии считаются «по умолчанию». Например, если вы ищете 1.6.0+CU102 Вы не найдете этого, это потому, что установка pip install.6.0 автоматически извлекивает Cuda’s 10.2 версия.

2. Если вы мазохист и нуждаетесь в версии только для процессора: PIP установка 1.6.0+процессор

3. Чтобы сохранить этот репо на ваш Требования.txt , просто добавьте это в качестве первой линии:

-f https://download.pytorch.org/whl/torch_stable.html

...
torch==1.6.0+cu110
...

Счастливая факел!

Оригинал: “https://dev.to/moniquelive/pre-compiled-pytorch-versions-5b7l”