Giters

tiangolo / uvicorn-gunicorn-machine-learning-docker

Docker image for high-performance Machine Learning web applications. With Uvicorn managed by Gunicorn in Python 3.7 and 3.6, using Conda, with CUDA and TensorFlow variants.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Will CUDA version be updated to 10.1 version?

Lelthar opened this issue · comments

commented

Hi @tiangolo

I want to use TensorFlow 2.2 using gpu in my project, but in a Tensorflow 2.2 project CUDA 10.1 is required. Will it be updated in the future?

Thanks

commented

Hey there! I'm sorry, I'm currently not using this Docker image, so I wouldn't be able to promise much. You can probably extract some bits and pieces from the current files in case that's useful, but I won't be able to work on it much any time soon.

commented

Assuming the original issue was solved, it will be automatically closed now. But feel free to add more comments or create new issues.