Ondrej Megela 2020-12-03 15:28:22 +00:00
parent cb6989009e
commit ea1d5d5a96

@ -19,10 +19,15 @@ conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
conda install pytorch torchvision torchaudio cpuonly -c pytorch
```
2. Ak chceme využiť GPU je potrebné nainštalovať pytorch tzv."from source"
najprv vhodnu verziu CUDA napr:
najprv vhodnú verziu CUDA napr:
```
conda install -c anaconda cudatoolkit=10.2
```
ďalej nainštalovať NCCL
```
conda install -c anaconda nccl
```
potom nainštalovať závislosti:
```
conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses