... | ... | @@ -8,13 +8,30 @@ The following example code will create a conda environment named shark-gpu which |
|
|
Installation of the tensorflow-gpu meta-module automatically takes care of dependencies such as the cuda libraries.
|
|
|
|
|
|
The setGPU module is installed (via pip as not conda package exists for it) to enable for automatic selection of a free GPU:
|
|
|
**Load the conda3 module to set the conda environment**
|
|
|
```
|
|
|
conda create -n shark-gpu python=3.6 # create a new conda environment
|
|
|
source activate shark-gpu # activate the environment
|
|
|
conda install cython tensorflow-gpu # install tensorflow
|
|
|
pip install setGPU # install setGPU
|
|
|
module load conda3
|
|
|
```
|
|
|
**Create a new conda environment in your data export directory**
|
|
|
```
|
|
|
conda create --prefix </exports/yourdataDIR>/shark-gpu python=3.6
|
|
|
```
|
|
|
**Activate the environment**
|
|
|
```
|
|
|
source activate </exports/yourdataDIR>/shark-gpu
|
|
|
```
|
|
|
**Install tensorflow**
|
|
|
```
|
|
|
conda install cython tensorflow-gpu
|
|
|
```
|
|
|
**Install setGPU**
|
|
|
```
|
|
|
pip install setGPU
|
|
|
```
|
|
|
**Note: to deactivate your conda environment**
|
|
|
```
|
|
|
source deactivate
|
|
|
```
|
|
|
|
|
|
## Accessing a GPU node
|
|
|
|
|
|
### Scheduling a GPU job using SGE
|
... | ... | |