Compare commits

..

1 Commits

Author SHA1 Message Date
1523e08a9e install quant docker 2024-11-21 15:27:59 +01:00
2 changed files with 2 additions and 2 deletions

View File

@ -3,7 +3,7 @@ name: Build docker images (scheduled)
on:
push:
branches:
- test_docker_run_quantization
- update-quantization-docker
repository_dispatch:
workflow_call:
inputs:

View File

@ -9,7 +9,7 @@ SHELL ["sh", "-lc"]
# The following `ARG` are mainly used to specify the versions explicitly & directly in this docker file, and not meant
# to be used as arguments for docker build (so far).
ARG PYTORCH='2.5.1'
ARG PYTORCH='2.4.1'
# Example: `cu102`, `cu113`, etc.
ARG CUDA='cu118'