Environment setup

#35

by pr-tet-usr - opened Apr 2

Apr 2

This is a great model and I was trying to finetune it for my specific use case, but somehow every run ends up in the same wheel- box
RuntimeError: CUDA driver error: operation not supported

Have tried various package version setups and nemo library setups, none of them seem to work.
Myself I have now Python 3.11.14 and Cuda version 13.0.

Can anyone provide suggestions for the setup that would allow to run the model train or maybe their package versions?

krirbs

Jun 5

Hi,
I've recently replicated my Conda environment for finetuning Parakeet. These versions and packages worked well for me

conda create -n nemo-ft python=3.12.9
conda activate nemo-ft
conda install pip
pip install torch==2.6.0+cu118 torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install transformers==4.51.3 librosa==0.11.0
pip install "nemo_toolkit[asr]==2.4.1"

Hope that helps! Regards

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment