Skip to content

tensorflow gpu version not enabled in okwrtdsh/anaconda3:keras-9.2-cudnn7 #1

@alidadsetan

Description

@alidadsetan

Running the container:
sudo nvidia-docker run -it okwrtdsh/anaconda3:keras-9.2-cudnn7 bash

"attaching" to container:
sudo docker exec -it container_name bash

Running python repl and importing tensorflow to check whether gpu is enabled in tensorflow:

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

which prints:

2018-07-05 14:56:47.988072: I tensorflow/core/platform/cpu_feature_guard.cc:140] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 3815053792300585653
]

That shows device_type as cpu. Can this be a bug?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions