Open DaemonSnake opened 1 year ago
added the following to the dockerfile
RUN pip install torchmetrics==0.11.4 torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchtext==0.14.1 torchaudio==0.13.1 torchdata==0.5.1 --extra-index-url https://download.pytorch.org/whl/cu117
works now
Hi The image crashes when starting the ui on a torchmetrics error