Skip to content

Conversation

gigony
Copy link
Collaborator

@gigony gigony commented Jan 24, 2022

By default, app packager is using nvcr.io/nvidia/pytorch:21.07-py3 as a base image and the image already has PyTorch that were manually compiled by NVIDIA to support latest architectures such as Amphere.

Since --upgrade was used in pip command to install necessary packages, pytorch is upgraded but with wring configuration (1.10.1+cu102) when installing PyTorch requires additional instructions for installation.

This patch removes the option when installing packages so that already-installed packages does not upgrade unless needed.

Signed-off-by: Gigon Bae gbae@nvidia.com

Signed-off-by: Gigon Bae <gbae@nvidia.com>
@gigony gigony added the bug Something isn't working label Jan 24, 2022
@gigony gigony added this to the v0.3.0 milestone Jan 24, 2022
@gigony gigony requested a review from MMelQin January 24, 2022 17:47
@gigony gigony self-assigned this Jan 24, 2022
@gigony gigony requested a review from KavinKrishnan January 24, 2022 17:47
Copy link
Collaborator

@MMelQin MMelQin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Look good to me for this one.

@gigony gigony merged commit 50b4cb9 into main Jan 25, 2022
@gigony gigony mentioned this pull request Jan 25, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants