Skip to content

Commit ba42f64

Browse files
fix: fix pytorch in the cuda11 image (#4841)
As reported by #4837 (comment), the pytorch in the cuda 11 image is currently cuda12. <!-- This is an auto-generated comment: release notes by coderabbit.ai --> ## Summary by CodeRabbit * **Chores** * Cleaned up documentation formatting by removing unnecessary blank lines for improved readability and consistency. * **Refactor** * Improved Docker installation process to better manage PyTorch backend selection for CUDA 11 environments. <!-- end of auto-generated comment: release notes by coderabbit.ai --> --------- Signed-off-by: Jinzhe Zeng <jinzhe.zeng@ustc.edu.cn> Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 5450066 commit ba42f64

3 files changed

Lines changed: 1 addition & 10 deletions

File tree

doc/model/show-model-info.md

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@ dp --pt show <INPUT> <ATTRIBUTES...>
1111

1212
- `<INPUT>`: Path to the model checkpoint file or frozen model file.
1313
- `<ATTRIBUTES>`: One or more information categories to display. Supported values are:
14-
1514
- `model-branch`: Shows available branches for multi-task models.
1615
- `type-map`: Shows the type mapping used by the model.
1716
- `descriptor`: Displays the model descriptor parameters.
@@ -34,35 +33,28 @@ dp show frozen_model.pth type-map descriptor fitting-net size
3433
Depending on the provided attributes and the model type, the output includes:
3534

3635
- **Model Type**
37-
3836
- Logs whether the loaded model is a _singletask_ or _multitask_ model.
3937

4038
- **model-branch**
41-
4239
- _Only available for multitask models._
4340
- Lists all available model branches and the special `"RANDOM"` branch, which refers to a randomly initialized fitting net.
4441

4542
- **type-map**
46-
4743
- For multitask models: Shows the type map for each branch.
4844
- For singletask models: Shows the model's type map.
4945

5046
- **descriptor**
51-
5247
- For multitask models: Displays the descriptor parameter for each branch.
5348
- For singletask models: Displays the descriptor parameter.
5449

5550
- **fitting-net**
56-
5751
- For multitask models: Shows the fitting network parameters for each branch.
5852
- For singletask models: Shows the fitting network parameters.
5953

6054
- **size**
61-
6255
- Prints the number of parameters for each component (`descriptor`, `fitting-net`, etc.), as well as the total parameter count.
6356

6457
- **observed-type**
65-
6658
- Displays the count and list of observed element types of the model during data statistics.
6759
- For multitask models, it shows the observed types for each branch.
6860
- Note: This info shows the types observed during training data statistics, which may differ from the type map.

doc/train/multi-task-training.md

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@ Specifically, there are several parts that need to be modified:
4848
- {ref}`model/model_dict <model/model_dict>`: The core definition of the model part and the explanation of sharing rules,
4949
starting with user-defined model name keys `model_key`, such as `my_model_1`.
5050
Each model part needs to align with the components of the single-task training {ref}`model <model>`, but with the following sharing rules:
51-
5251
- If you want to share the current model component with other tasks, which should be part of the {ref}`model/shared_dict <model/shared_dict>`,
5352
you can directly fill in the corresponding `part_key`, such as
5453
`"descriptor": "my_descriptor", `

source/install/docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ ENV PATH="/opt/deepmd-kit/bin:$PATH"
88
ENV VIRTUAL_ENV="/opt/deepmd-kit"
99
# Install package
1010
COPY dist /dist
11-
RUN if [ "${CUDA_VERSION}" = 11 ]; then uv pip install torch --index-url https://download.pytorch.org/whl/cu118; fi \
11+
RUN if [ "${CUDA_VERSION}" = 11 ]; then export UV_TORCH_BACKEND=cu118; fi \
1212
&& uv pip install "$(ls /dist/deepmd_kit${VARIANT}-*manylinux*_x86_64.whl)[gpu,cu${CUDA_VERSION},lmp,ipi,torch]" \
1313
&& dp -h \
1414
&& lmp -h \

0 commit comments

Comments
 (0)