Use Pytorch CPU

Dockerfile

@@ -1,24 +1,3 @@
-#!/usr/bin/env bash
-# Copyright (c) 2023, NVIDIA CORPORATION.  All rights reserved.
-#
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-#
-#     http://www.apache.org/licenses/LICENSE-2.0
-#
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-# Use this script to install KenLM, OpenSeq2Seq decoder, Flashlight decoder, OpenGRM Ngram tool to contaner 
-
-# How to use? Build it from NeMo root folder:
-# 1. git clone https://github.com/NVIDIA/NeMo.git && cd NeMo
-# 2. DOCKER_BUILDKIT=1 docker build -t nemo:23.03.1 -f ./scripts/installers/Dockerfile.ngramtools .
-
 FROM python:3.11
 
 RUN --mount=target=/tmp/packages.txt,source=packages.txt 	apt-get update && 	xargs -r -a /tmp/packages.txt apt-get install -y && 	rm -rf /var/lib/apt/lists/*
@@ -27,12 +6,10 @@ RUN --mount=target=/tmp/pre-requirements.txt,source=pre-requirements.txt 	pip in
 
 WORKDIR /workspace
 
-RUN git clone https://github.com/NVIDIA/NeMo nemo
+RUN wget https://github.com/NVIDIA/NeMo/archive/refs/tags/r2.0.0rc1.zip && unzip r2.0.0rc1.zip && mv NeMo-r2.0.0rc1 nemo
 
 WORKDIR /workspace/nemo
 
-RUN git checkout r2.0.0rc1
-
 RUN ./reinstall.sh
 
 RUN --mount=target=/tmp/requirements.txt,source=requirements.txt     pip install --no-cache-dir -r /tmp/requirements.txt

pre-requirements.txt

@@ -1,2 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
 Cython
-packaging
+packaging
+torch==2.5.1+cpu

requirements.txt

@@ -1,7 +1,5 @@
-# Cython
-torch
-huggingface-hub==0.23.2
 # nemo-toolkit[asr]==2.0.0rc1
+huggingface-hub==0.23.2
 numpy<2.0.0
 ipython
 # yt_dlp