mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-08-01 11:12:50 +08:00
[docker] upgrade to torch 2.6 (#7442)
This commit is contained in:
parent
c841e92116
commit
747e02d60d
@ -1,6 +1,6 @@
|
||||
# Default use the NVIDIA official image with PyTorch 2.3.0
|
||||
# Default use the NVIDIA official image with PyTorch 2.6.0
|
||||
# https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/index.html
|
||||
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.02-py3
|
||||
ARG BASE_IMAGE=nvcr.io/nvidia/pytorch:24.12-py3
|
||||
FROM ${BASE_IMAGE}
|
||||
|
||||
# Define environments
|
||||
|
Loading…
x
Reference in New Issue
Block a user