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