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 conda install pytorch torchvision torchaudio cpuonly -c pytorch
``` ```
2. Ak chceme využiť GPU je potrebné nainštalovať pytorch tzv."from source" 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 conda install -c anaconda cudatoolkit=10.2
``` ```
ďalej nainštalovať NCCL
```
conda install -c anaconda nccl
```
potom nainštalovať závislosti: potom nainštalovať závislosti:
``` ```
conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses conda install numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses