search

ufoym / deepo

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

PyTorch compiled with cuda and cudnn, so we don't need to build docker based on nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04. #113

Closed ypw-rich closed 2 years ago

ypw-rich commented 5 years ago

PyTorch compiled with cuda and cudnn, so we don't need to build docker based on nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04.

cuda:10.0-base-ubuntu18.04 is only 115MB，but cuda:10.0-cudnn7-devel-ubuntu18.04 is 3.05GB.

ufoym commented 5 years ago

cuda:10.0-base-ubuntu18.0 contains only the bare minimum (libcudart), while most DL frameworks depends on more CUDA packages?

ypw-rich commented 5 years ago

Because PyTorch already compiled with cuda, it doesn't load cuda in the system, so we can add a if else to PyTorch to avoid overpacking.

ufoym commented 5 years ago

@ypw-rich Thanks for the suggestion. Could you fix this and make a pull request? We will merge the code and trigger a rebuild for the related images.

github-actions[bot] commented 2 years ago

This issue is stale because it has been open for 180 days with no activity.

github-actions[bot] commented 2 years ago

This issue was closed because it has been inactive for 30 days since being marked as stale.