ros/LLaMA-Factory
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[assets] fix docker images (#8203)

Browse Source
This commit is contained in:
hoshi-hiyouga
2025-05-28 22:26:05 +08:00
committed by GitHub
parent 519bab86e6
commit 83a9ff5853
13 changed files with 56 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/docker-cuda/Dockerfile.base
View File

@@ -1,6 +1,6 @@
# Start from the pytorch official image (ubuntu-22.04 + cuda-12.4.1 + python-3.11)
# https://hub.docker.com/r/pytorch/pytorch/tags
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-runtime
FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
# Define environments
ENV MAX_JOBS=16