Skip to content
This repository was archived by the owner on Aug 3, 2021. It is now read-only.
This repository was archived by the owner on Aug 3, 2021. It is now read-only.

GPU is not being used? #562

@Ramnathan-zoimeet

Description

@Ramnathan-zoimeet

I did follow the instruction as per requirement i installed CUDA 10.2, and TensorFlow - 1.15.4 and Python 3.6.9 and to make sure all installation went right i did run the run_all_tests.sh and it was successful with no errors, but still when i run the main script with train/ infer/ eval any of these modes the GPU is being assigned right but it does not use it or speed up the process (using a single 16gb v100 GPU and tried with horovod ), so is there anything else i need to do to make the GPU work?

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions