github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 20:39:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Upgrade torch version to 2.2.1 (#1374)

Browse Source 
Co-authored-by: yyhhyy <yyhhyyyyyy@163.com>
Co-authored-by: Fangyin Cheng <staneyffer@gmail.com>
This commit is contained in:
yyhhyy
2024-04-08 18:35:19 +08:00
committed by GitHub
parent bb77e13dee
commit 5a0c20a864
3 changed files with 103 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/base/Dockerfile
View File

@@ -1,4 +1,4 @@
ARG BASE_IMAGE="nvidia/cuda:11.8.0-runtime-ubuntu22.04"
ARG BASE_IMAGE="nvidia/cuda:12.1.0-runtime-ubuntu22.04"
FROM ${BASE_IMAGE}
ARG BASE_IMAGE