Differences
This shows you the differences between two versions of the page.
| Both sides previous revision Previous revision Next revision | Previous revision | ||
|
en:services:application_services:high_performance_computing:tensorflow [2020/03/27 09:11] akhuziy [Installing TensorFlow] |
en:services:application_services:high_performance_computing:tensorflow [2020/03/27 09:58] (current) akhuziy [Installing TensorFlow] |
||
|---|---|---|---|
| Line 5: | Line 5: | ||
| It is recommended to use the python tools [[https://virtualenv.pypa.io/en/latest/user_guide.html|virtualenv]] and [[https://pip.pypa.io/en/stable/quickstart/|pip]] to create a virtual python environment and install the desired version of tensorflow within that environment. | It is recommended to use the python tools [[https://virtualenv.pypa.io/en/latest/user_guide.html|virtualenv]] and [[https://pip.pypa.io/en/stable/quickstart/|pip]] to create a virtual python environment and install the desired version of tensorflow within that environment. | ||
| <code> | <code> | ||
| - | module load python/3.5.1 | + | module load python/3.6.3 |
| virtualenv <your_virtual_env> | virtualenv <your_virtual_env> | ||
| cd <your_virtual_env> | cd <your_virtual_env> | ||
| source bin/activate | source bin/activate | ||
| pip install --upgrade pip | pip install --upgrade pip | ||
| - | pip install tensorflow-gpu==1.12.0 | + | pip install tensorflow-gpu==1.15.0 |
| </code> | </code> | ||
| - | If you do not want to use GPUs simply replace the last line with <code>pip install tensorflow==1.12.0</code> | + | If you do not want to use GPUs simply replace the last line with <code>pip install tensorflow==1.15.0</code> |
| ==== Testing the installation ==== | ==== Testing the installation ==== | ||
| Line 20: | Line 20: | ||
| #SBATCH -p gpu | #SBATCH -p gpu | ||
| #SBATCH -t 1 | #SBATCH -t 1 | ||
| - | #SBATCH --gres=gpu:1 | + | #SBATCH --gpus-per-node 1 |
| - | + | ||
| - | module load cuda90/toolkit/9.0.176 | + | module load cuda10.0/toolkit/10.0.130 |
| - | module load cuda90/blas/9.0.176 | + | module load cuda10.0/blas/10.0.130 |
| - | module load cudnn/90v7.3.1 | + | module load cudnn/10.0v7.6.3 |
| python tftest.py | python tftest.py | ||