Differences
This shows you the differences between two versions of the page.
Both sides previous revision Previous revision Next revision | Previous revision Next revision Both sides next revision | ||
gpu [2021/12/15 17:53] kasner Update CUDA version |
gpu [2021/12/15 17:57] kasner [Set-up CUDA and CUDNN] |
||
---|---|---|---|
Line 67: | Line 67: | ||
fi | fi | ||
- | * When not using Theano, just Tensorflow this can be simplified to '' | + | * When not using Theano, just Tensorflow this can be simplified to '' |
* Note that the '' | * Note that the '' | ||
Line 89: | Line 89: | ||
This environment have TensorFlow 1.8.0 and all necessary requirements for NeuralMonkey. | This environment have TensorFlow 1.8.0 and all necessary requirements for NeuralMonkey. | ||
- | ==== Pytorch | + | ==== PyTorch |
- | If you want to use pytorch, there is a ready-made environment in | + | Install PyTorch following the instructions on https://pytorch.org. |
- | / | + | At the time of writing, the recommended |
- | + | ||
- | It does rely on the CUDA and CuDNN setup above. | + | |
+ | pip3 install torch==1.10.0+cu111 torchvision==0.11.1+cu111 torchaudio==0.10.0+cu111 -f https:// | ||
==== Using cluster ==== | ==== Using cluster ==== | ||