ufoym / deepo

Setup and customize deep learning environment in seconds.

Home Page:http://ufoym.com/deepo

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to upgrade cuda for the image?

chenwydj opened this issue · comments

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!

commented

You can specify cuda version in the following line:

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 :)