Comments (5)
Hy @eyacov!
I am not sure about the Python version compatibility.
It was tested by Python 3.6 and Nvidia's docker image from [nvcr.io/nvidia/pytorch:20.09-py3].(https://catalog.ngc.nvidia.com/orgs/nvidia/containers/pytorch)
Could you provide the error messages you're encountering?
from nuwave.
Hey @junjun3518
I can't use this container because it's not available for my GPU: GeForce RTX 4090
when I try to run it without a container I'm getting the following error
Traceback (most recent call last): File "trainer.py", line 139, in <module> train(args) File "trainer.py", line 126, in train trainer.fit(model) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 510, in fit results = self.accelerator_backend.train() File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/accelerators/accelerator.py", line 57, in train return self.train_or_test() File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/accelerators/accelerator.py", line 74, in train_or_test results = self.trainer.train() File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 532, in train self.run_sanity_check(self.get_model()) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 730, in run_sanity_check _, eval_results = self.run_evaluation(max_batches=self.num_sanity_val_batches) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 646, in run_evaluation output = self.evaluation_loop.evaluation_step(batch, batch_idx, dataloader_idx) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/trainer/evaluation_loop.py", line 180, in evaluation_step output = self.trainer.accelerator_backend.validation_step(args) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/accelerators/gpu_accelerator.py", line 73, in validation_step return self._step(self.trainer.model.validation_step, args) File "/home/etay/anaconda3/envs/nuwave/lib/python3.6/site-packages/pytorch_lightning/accelerators/gpu_accelerator.py", line 65, in _step output = model_step(*args) File "/media/etay/Daten/localization/nuwave/lightning_model_bird.py", line 239, in validation_step 0, self.max_step, (wav.shape[0], ), device=self.device) + 1 RuntimeError: CUDA error: no kernel image is available for execution on the device CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
from nuwave.
Hi @eyacov ,
It seems that error message is about cuda version issue.
Since this project started at 2020, it is utilizing old version of torch and cuda, which do not support 4090.
For now, I do not have any resource to test and do not have authorization to change this repo.
I recommend to use recent base image such as nvcr.io/nvidia/pytorch:23.07-py3 (I am using it for 4090 now).
from nuwave.
hi @junjun3518
Trying to run the code this docker leads to the following error
Traceback (most recent call last):
File "/media/etay/Daten/localization/nuwave/trainer.py", line 1, in
from lightning_model import NuWave
File "/media/etay/Daten/localization/nuwave/lightning_model.py", line 9, in
import pytorch_lightning as pl
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/init.py", line 66, in
from pytorch_lightning import metrics
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/metrics/init.py", line 14, in
from pytorch_lightning.metrics.metric import Metric
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/metrics/metric.py", line 23, in
from pytorch_lightning.metrics.utils import _flatten, dim_zero_cat, dim_zero_mean, dim_zero_sum
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/metrics/utils.py", line 18, in
from pytorch_lightning.utilities import rank_zero_warn
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/init.py", line 24, in
from pytorch_lightning.utilities.apply_func import move_data_to_device
File "/usr/local/lib/python3.10/dist-packages/pytorch_lightning/utilities/apply_func.py", line 25, in
from torchtext.data import Batch
ImportError: cannot import name 'Batch' from 'torchtext.data' (/usr/local/lib/python3.10/dist-packages/torchtext/data/init.py)
Do I need to change something in requirements.txt to make this work?
from nuwave.
I manged to solve the issue. The requirements file needs to be written as follows:
ffmpeg
torchtext==0.6.0
pytorch_lightning==1.1.6
prefetch_generator
librosa==0.8.0
omegaconf==2.0.6
You might want to consider changing it
from nuwave.
Related Issues (14)
- MU-GAN outputs HOT 2
- . HOT 1
- Upsample file has static in the background or complete silence HOT 2
- About pretrained model weight HOT 1
- How do I upsample a wav once the model is trained? HOT 3
- _pickle.UnpicklingError running test.py script (SOLVED) HOT 1
- robustness issue: use of methods marked for deprecation
- pretrained weight HOT 1
- Contribution: checkpoints AVAILABLE! HOT 8
- About to be a Contribution HOT 10
- Whether sampling.py in nuwave can be used in nuwave2
- Pretrained model HOT 4
- Different sampling rates HOT 26
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from nuwave.