mirror of
https://github.com/clearml/clearml-fractional-gpu
synced 2025-03-16 10:28:55 +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
|
||
|
|
|