diff --git a/docker/build_image.py b/docker/build_image.py index b54b7440..a9e1a107 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -23,17 +23,17 @@ class Builder: # A mirrored image of nvidia/cuda:12.4.0-devel-ubuntu22.04 args.base_image = 'nvidia/cuda:12.1.0-devel-ubuntu22.04' if not args.torch_version: - args.torch_version = '2.3.0' - args.torchaudio_version = '2.3.0' - args.torchvision_version = '0.18.0' + args.torch_version = '2.3.1' + args.torchaudio_version = '2.3.1' + args.torchvision_version = '0.18.1' if not args.tf_version: args.tf_version = '2.16.1' if not args.cuda_version: args.cuda_version = '12.1.0' if not args.vllm_version: - args.vllm_version = '0.5.1' + args.vllm_version = '0.5.3' if not args.lmdeploy_version: - args.lmdeploy_version = '0.5.0' + args.lmdeploy_version = '0.6.2' if not args.autogptq_version: args.autogptq_version = '0.7.1' return args