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
|
||||||
|
Loading…
Reference in New Issue
Block a user