starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add Docker image for CUDA-12.1.

Browse Source
This commit is contained in:
苏阳
2024-01-08 14:22:05 +08:00
parent 4aab1d490b
commit 23a01b0696
10 changed files with 146 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
README_CN.md
View File

@@ -1040,6 +1040,7 @@ print(response.choices[0].message.content)
1. 根据需要使用的镜像版本，安装相应版本的Nvidia驱动：
- `qwenllm/qwen:cu117`（**推荐**）：`>= 515.48.07`
- `qwenllm/qwen:cu114`（不支持flash-attention）：`>= 470.82.01`
- `qwenllm/qwen:cu121`：`>= 530.30.02`
- `qwenllm/qwen:latest`：与`qwenllm/qwen:cu117`相同
2. 安装并配置[docker](https://docs.docker.com/engine/install/)和[nvidia-container-toolkit](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html)：