search

mrdbourke / tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.
https://dbourke.link/ZTMTFcourse
MIT License
5.14k stars 2.53k forks source link

What version of TensorFlow to install locally with Ubuntu 20.04 and GPU RTX3060? #505

Closed sgkouzias closed 1 year ago

sgkouzias commented 1 year ago

I have a GPU enabled PC (RTX3060) with Ubuntu 20.04 installed. I still can't figure what version of TensorFlow could I use to complete the first milestone project (food vision based on the respective template) ! I tried TensorFlow version 2.11 but I could not save and load the model. I tried TensorFlow version 2.9.1 but it could not utilize mixed precision. I tried TensorFlow version 2.4.1 but it could not fit the model.. Is a requirements.txt available in order to use create a TensorFlow environment locally and complete the project?

sgkouzias commented 1 year ago

Ultimately I installed TensorFlow==2.8.1 and it worked. Everything now is ok (mixed precision utilized, model saved and reloaded).