From d7b85e03dcbdb50394490a86b0a07fe0a1b290e4 Mon Sep 17 00:00:00 2001 From: Herbert Kruitbosch Date: Thu, 18 Jan 2018 09:01:01 +0000 Subject: [PATCH] Update 'README.md' --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 62b325b..7322ddb 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,5 @@ # GPU / CPU Benchmarks for tensorflow and keras -The benchmarks run on all the devices that tensorflow finds. If that should include a GPU, make sure to install the python egg `tensorflow-gpu`, also make sure the nvidia kernel module has the version supported by your cuda installation, in my case I needed the ubuntu-package `nvidia-384==384.111-0ubuntu1` (the 111-part also has to match). You'll see an error message in the **console** (not the notebook) when importing tensorflow and the versions mismatch. Also make sure cuda is in your `LD_LIBRARY_PATH`, in my case `/usr/local/cuda-8.0/targets/x86_64-linux/lib/` for the ubuntu package `cuda-8.0`, the folder should contain object files like `libcudnn.so` ('updatedb' and `locate libcuda.so` to find such a folder on your linux system). +The benchmarks run on all the devices that tensorflow finds. If that should include a GPU, make sure to install the python egg `tensorflow-gpu`, also make sure the nvidia kernel module has the version supported by your cuda installation, in my case I needed the ubuntu-package `nvidia-384==384.111-0ubuntu1` (the 111-part also has to match). You'll see an error message in the **console** (not the notebook) when importing tensorflow and the versions mismatch. Also make sure cuda is in your `LD_LIBRARY_PATH`, in my case `/usr/local/cuda-8.0/targets/x86_64-linux/lib/` for the ubuntu package `cuda-8.0`, the folder should contain object files like `libcudnn.so` (`updatedb` and `locate libcuda.so` to find such a folder on your linux system). Run the benchmarks preferrably in a virtualenv with `python>=3.4,<4`