search

InseeFrLab / images-datascience

Collection of Docker images to build the data science catalog of the Onyxia project
MIT License
24 stars 22 forks source link

Pytorch : pin pytorch cuda version #142

Closed olevitt closed 11 months ago

olevitt commented 11 months ago

Hi !

At the moment when installing torch we don't specify any cuda version which results in cuda 12 libraries although we are on base 11. See https://github.com/InseeFrLab/images-datascience/blob/34d7d6acf3a087468415c627d658d51636a716b4/python-pytorch/Dockerfile#L10C26-L10C26
We should consider pinning the cuda version using something like : 

pip install torch --force-reinstall --index-url https://download.pytorch.org/whl/cu118
pip install torchvision --force-reinstall --index-url https://download.pytorch.org/whl/cu118
avouacr commented 11 months ago

Thank's, done in 32d19216e9a07076f12ac110cf02d73f53d1ebe5