mirror of
https://github.com/hiyouga/LLaMA-Factory.git
synced 2025-10-14 15:52:49 +08:00
[docker] upgrade to torch 2.6 (#7442)
This commit is contained in:
parent
3aa4f32e9c
commit
3612946dd9
@ -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