How to upgrade cuda for the image?

[chenwydj]

I started using deepo, and on top of that I installed more packages, and then commit and pushed to my own dockerhub as another version.

What is the best practice to upgrade the cuda version for the image?

Thank you very much!

[ufoym]

You can specify cuda version in the following line:

deepo/scripts/make-gen-docker.py

Line 82 in 4004501

generate(f, '11.1', '8')

If what you wanted is to upgrade the cuda version to 11.1, the latest deepo images should meet your requirement :)