Python で機械学習のベースイメージ、何選ぶ？
#Python
やっぱり CUDA がないと始まらない
nvidia/cuda - Docker Image | Docker Hub
タグは cuDNN まで全部入りの
-cudnn-devel-ubuntu24.04
が無難そう
あるいは
PyTorch | NVIDIA NGC
CUDA Toolkit と PyTorch が同梱済で便利
いちいち CUDA のバージョンを合わせる作業をしなくて済むのでよさそう
ちょっとパッケージの更新が遅いけどこれも
Deep Learning Containers のドキュメント | Google Cloud