Can I install miniconda in the image nvcr.io/nvidia/pytorch:23.08-py3 and use PyTorch 2.1.0a0+29c30b1 installed in that image in my conda environment?

Description

I know that conda is installed as far as the nvcr.io/nvidia/pytorch:22.10-py3 image. However, conda has been excluded from later versions. I use nvcr.io/nvidia/pytorch:23.08-py3 image for training. However, the lack of conda in that image makes it inconvenient to configure various experimental environments. So I installed miniconda, but I can’t use pytorch 2.1.0a0+29c30b1, which is already installed in that image. Is there any way to use pytorch 2.1.0a0+29c30b1, which is already installed in the image, in a conda environment?

Environment

TensorRT Version:
GPU Type: A100
Nvidia Driver Version: 535.129.03
CUDA Version: 12.2
CUDNN Version: 8.8.1
Operating System + Version: Linux RHEL 8.8
Python Version (if applicable): 3.10.12
TensorFlow Version (if applicable):
PyTorch Version (if applicable): 2.1.0a0+29c30b1
Baremetal or Container (if container which image + tag): nvcr.io/nvidia/pytorch:23.08-py3

Relevant Files

Please attach or include links to any models, data, files, or scripts necessary to reproduce your issue. (Github repo, Google Drive, Dropbox, etc.)

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered

Hi @hopangbear ,
This forum talks about issues specific to TRT. I am afraid i might not be able to help you on the topic.