search

ufoym / deepo

Setup and customize deep learning environment in seconds.
http://ufoym.com/deepo
MIT License
6.32k stars 749 forks source link

How to upgrade cuda for the image? #143

Closed chenwydj closed 2 years ago

chenwydj commented 3 years ago

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 commented 2 years ago

You can specify cuda version in the following line: https://github.com/ufoym/deepo/blob/40045012059ffd298a08fdb5e9f95436e7015642/scripts/make-gen-docker.py#L82

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