Upgrade tensorflow version to support higher versions of CUDA

Open kimjson opened this issue 2 years ago • 3 comments

Current

tensorflow version is 2.3.*, which is compatible with CUDA 10.1 (not 11) according to: https://www.tensorflow.org/install/source#gpu.

tensorflow version is updated to 2.4.* or more to support CUDA version 11.* or more.

Dec 11 '21 08:12 kimjson

Hi @sundoufu ,

Thanks for the suggestion! I've updated the requirements file.

Best, Yu

Dec 11 '21 18:12 yumeng5

Hi @sundoufu ,

Thanks for the suggestion! I've updated the requirements file.

Best, Yu

Thanks for the fast response!! Is it also working for the higher tensorflow version without any changes?

Dec 12 '21 15:12 kimjson

Yes, it should work without changing any code.

Dec 13 '21 02:12 yumeng5