github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-01 16:18:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix docker build error

Browse Source
This commit is contained in:
FangYin Cheng 2023-08-23 10:56:37 +08:00
parent 1b9bb47ee2
commit 95fce1bab2
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
docker/base/build_image.sh
View File

@ -4,7 +4,7 @@ SCRIPT_LOCATION=$0
cd "$(dirname "$SCRIPT_LOCATION")" cd "$(dirname "$SCRIPT_LOCATION")"
WORK_DIR=$(pwd) WORK_DIR=$(pwd)
BASE_IMAGE="nvidia/cuda:11.8.0-devel-ubuntu22.04" BASE_IMAGE="nvidia/cuda:11.8.0-runtime-ubuntu22.04"
IMAGE_NAME="eosphorosai/dbgpt" IMAGE_NAME="eosphorosai/dbgpt"
# zh: https://pypi.tuna.tsinghua.edu.cn/simple # zh: https://pypi.tuna.tsinghua.edu.cn/simple
PIP_INDEX_URL="https://pypi.org/simple" PIP_INDEX_URL="https://pypi.org/simple"
@ -14,7 +14,7 @@ BUILD_LOCAL_CODE="false"
LOAD_EXAMPLES="true" LOAD_EXAMPLES="true"
usage () { usage () {
echo "USAGE: $0 [--base-image nvidia/cuda:11.8.0-devel-ubuntu22.04] [--image-name db-gpt]" echo "USAGE: $0 [--base-image nvidia/cuda:11.8.0-runtime-ubuntu22.04] [--image-name db-gpt]"
echo " [-b|--base-image base image name] Base image name" echo " [-b|--base-image base image name] Base image name"
echo " [-n|--image-name image name] Current image name, default: db-gpt" echo " [-n|--image-name image name] Current image name, default: db-gpt"
echo " [-i|--pip-index-url pip index url] Pip index url, default: https://pypi.org/simple" echo " [-i|--pip-index-url pip index url] Pip index url, default: https://pypi.org/simple"

4
setup.py
View File

@ -216,8 +216,8 @@ def torch_requires(
cuda_version = get_cuda_version() cuda_version = get_cuda_version()
if not cuda_version: if not cuda_version:
torch_pkgs = [ torch_pkgs = [
f"torch=={torch_version}+cpu", f"torch=={torch_version}",
f"torchvision=={torchvision_version}+cpu", f"torchvision=={torchvision_version}",
f"torchaudio=={torchaudio_version}", f"torchaudio=={torchaudio_version}",
] ]
else: else: