mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-02 11:42:49 +08:00
Update Dockerfile
Former-commit-id: 5e8a619d1d20048ad912a1f6d9ccdae5740811bc
This commit is contained in:
parent
1ab3ea21ce
commit
5ad86fc71d
@ -1,3 +1,5 @@
|
||||
# Default use the NVIDIA official image with PyTorch 2.3.0
|
||||
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html
|
||||
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3
|
||||
FROM ${BASE_IMAGE}
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user