18 lines
624 B
Plaintext
18 lines
624 B
Plaintext
# tested in python==3.6.6
|
|
numpy==1.14.5
|
|
tensorflow==1.10.0 # to do... compatible with latest tf and tf-gpu
|
|
tensorboard==1.10.0
|
|
networkx==2.2
|
|
gensim==3.0.1
|
|
scikit-learn==0.19.0 # to do... compatible with >0.20
|
|
pandas==0.23.0
|
|
|
|
# Enable GPU:
|
|
# If using anaconda, run `conda install tensorflow-gpu==1.10.0`
|
|
|
|
# For cuda driver compatibility: https://docs.nvidia.com/deploy/cuda-compatibility/index.html
|
|
# For example, if driver version is 384.xx then also install cudatoolkit==9.0
|
|
|
|
# Or simply build from docker image: docker pull tensorflow/tensorflow:1.10.0-gpu-py3
|
|
# ref: https://www.tensorflow.org/install/docker#gpu_support
|