-
Notifications
You must be signed in to change notification settings - Fork 0
Home
Manuel edited this page Jun 12, 2019
·
3 revisions
The provided code works on Ubuntu 18.04 with the following configuration inside a virtualenv:
- Python 3.5.x: https://www.python.org/downloads/release/python-356/ (Compiled from source)
- Tensorflow-gpu >= 1.12: https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.13.1-cp35-cp35m-linux_x86_64.whl (installed via pip, pointing to that wheel)
- Keras: 2.2.4 (installed via pip)
- opencv-python >= 3.x
- CUDA 10 (needed by TF): https://www.tensorflow.org/install/gpu (steps: https://www.tensorflow.org/install/gpu#ubuntu_1804_cuda_10)