mirror of
https://github.com/clearml/clearml-fractional-gpu
synced 2025-06-26 18:26:02 +00:00
6 lines
187 B
Docker
6 lines
187 B
Docker
FROM allegroai/clearml-fractional-gpu:u22.04-cu12.3-8gb
|
|
|
|
# upgrade torch to the latest version
|
|
RUN pip3 install -U torch torchvision torchaudio torchdata torchmetrics torchrec torchtext
|
|
|