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/03 17:28] vodrazka |
gpu [2021/12/15 17:57] kasner [Pytorch Environment] |
||
---|---|---|---|
Line 55: | Line 55: | ||
You need to set library path from your '' | You need to set library path from your '' | ||
- | CUDA_version=10.1 | + | CUDA_version=11.1 |
- | CUDNN_version=7.6 | + | CUDNN_version=8.0 |
CUDA_DIR_OPT=/ | CUDA_DIR_OPT=/ | ||
if [ -d " | if [ -d " | ||
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 ==== | ||