Lightning-AI / pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

Home Page:https://lightning.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

docker image doesn't have `pytorch_lightning`

grisaitis opened this issue · comments

Bug description

$ docker run --rm pytorchlightning/pytorch_lightning:base-cuda-py3.10-torch2.2-cuda12.1.0 python -c "import pytorch_lightning"

==========
== CUDA ==
==========

CUDA Version 12.1.0

...

Traceback (most recent call last):
  File "<string>", line 1, in <module>
ModuleNotFoundError: No module named 'pytorch_lightning'

is this the intended behavior?

What version are you seeing the problem on?

v2.2

How to reproduce the bug

No response

Error messages and logs

No response

Environment

No response

More info

No response