[ Skip to the content ]

Institute of Formal and Applied Linguistics Wiki


[ Back to the navigation ]

Differences

This shows you the differences between two versions of the page.

Link to this comparison view

Both sides previous 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 [Pytorch Environment]
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 Environment ====+==== PyTorch Environment ====
  
-If you want to use pytorch, there is a ready-made environment in+Install PyTorch following the instructions on https://pytorch.org.
  
-  /home/hajicj/anaconda3/envs/pytorch/bin +At the time of writing, the recommended setup for CUDA 11.1 (supported by all GPU nodes) is:
-   +
-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://download.pytorch.org/whl/cu111/torch_stable.html
 ==== Using cluster ==== ==== Using cluster ====
  

[ Back to the navigation ] [ Back to the content ]