Skip to content
This repository has been archived by the owner on Jan 5, 2023. It is now read-only.

Commit

Permalink
README: add new backend info
Browse files Browse the repository at this point in the history
  • Loading branch information
ozancaglayan committed Feb 6, 2017
1 parent f3eb28e commit 6470257
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,8 @@ So if you care (and you absolutely should) about reproducibility, make sure your

## Configuring Theano

Here is a basic `.theanorc` file recommended for best performance:
Here is a basic `.theanorc` file (Note that the way you install CUDA, CuDNN
may require some modifications):

```
[global]
Expand All @@ -110,10 +111,6 @@ floatX = float32
# Keep theano compilation in RAM if you have a 7/24 available server
base_compiledir=/tmp/theano-%(user)s
[nvcc]
# This is already disabled by upstream Theano as well
fastmath = False
[cuda]
# CUDA 8.0 is better
root = /opt/cuda-7.5
Expand All @@ -129,6 +126,17 @@ include_path = /opt/CUDNN/cudnn-v5.1/include
cnmem = 0.95
```

If you have a recent Theano, you may want to try the new GPU backend after
installing [libgpuarray](https://github.com/Theano/libgpuarray). In order to do so,
pass `GPUARRAY=1` into the environment when running `nmt-train`:

```
$ GPUARRAY=1 nmt-train -c <conf file> ...
```

Note that we could not obtain accurate results using Maxwell GPUs with this backend
so use it at your own risk.

### Checking BLAS configuration

Recent Theano versions can automatically detect correct MKL flags. You should obtain a similar output after running the following command:
Expand Down

0 comments on commit 6470257

Please sign in to comment.