image-segmentation-keras icon indicating copy to clipboard operation
image-segmentation-keras copied to clipboard

Published 20 hours ago • verified publisher icon divamgupta

Cuda 11

Open quentingllmt opened this issue 4 years ago • 2 comments

Hi, I'm trying to run this repository on an NVIDIA Ampere GPU (RTX 3090).

This GPU needs Cuda 11.x which needs tensorflow 2.4.

Is the project compatible as keras is now part of tensorflow ? As anyone already done this ?

quentingllmt avatar Jan 12 '21 11:01 quentingllmt

Yes this works with CUDA 11.x with tensorflow 2.4. I have done it

Marius-Juston avatar Jan 20 '21 05:01 Marius-Juston

So you installed CUDA 11.x with tensorflow-gpu==2.4.1? What are the versions of the rest packages in order to run properly the image-segmnentation-keras? I am trying with an RTX 30xx, CUDA 11.x and tensorflow-gpu==2.4.1 but it doesn't recognize my GPU.

sotomotocross avatar Jan 20 '22 16:01 sotomotocross