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 |