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