Megatron Container Image Setting
Hi. I want to use 'nvcr.io/nvidia/pytorch:24.03-py3' image for using the megatron.
My start command is 'docker run --gpus all -it --rm -v /:/workspace/megatron -v /:/workspace/dataset -v /:/workspace/checkpoints nvcr.io/nvidia/pytorch:24.03-py3'
However, I have trouble with starting the pod.
How to fix the command?
My start command is 'docker run --gpus all -it --rm -v /:/workspace/megatron -v /:/workspace/dataset -v /:/workspace/checkpoints nvcr.io/nvidia/pytorch:24.03-py3'
However, I have trouble with starting the pod.
How to fix the command?
