Skip to content
This repository was archived by the owner on Nov 13, 2024. It is now read-only.
This repository was archived by the owner on Nov 13, 2024. It is now read-only.

Change tensorflow-gpu to tensorflow #5368

Description

@Minkiu

Hey there,

Wanted to check with you first, currently in requirements-cuda.txt we find tensorflow-gpu=2.4.0 which if I am not mistaken is legacy, and tensorflow=2.4.0 would just work as well, this is since tensorflow > 2.

I've wasted some time trying to set this up on linux, and this was one of the things that I had to change to make it work (among others), actually I used tensorflow without an specific version (installed 2.5.0) and so far so good, so it might be able to be bumped.

Let me know if you wnat a PR.

Cheers.

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