Tensorflow-GPU y Jupyter notebook en el cluster
matgreco opened this issue · comments
Dejo esto acá en caso de que a alguien le sirva.
El cluster cuenta con
- Cuda 8
- CuDNN 5
Por ende, para instalar tensorflow gpu deben hacerlo en un venv a través de pip. Debe ser la version 1.2.0
pip install tensorflow-gpu==1.2.0
Si al tratar de correr algo con tensorflow-gpu no les reconoce ninguna gpu, o les tira un error que no encuentra libcusolver.so, deben configurar la variable de entorno LD_PATH_LIBRARY de la siguiente forma (gracias a Rodolfo Palma):
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:/usr/local/cuda-8.0/lib64/::/opt/OpenBLAS/lib/:/opt/OpenBLAS/lib
Para jupyter notebook, cada snippet de código que hacen corre en un subproccess, por ende el procedimiento para configurar el path es otro y lo pueden encontrar acá