search

ben-hayes / neural-waveshaping-synthesis

efficient neural audio synthesis in the waveform domain
Mozilla Public License 2.0
185 stars 14 forks source link

Will this work with a newer version of PyTorch than 1.7.1? #8

Open znmeb opened 2 years ago

znmeb commented 2 years ago

I'm trying to install Neural Waveshaping Synthesis on an NVIDIA Jetson (arm64 with NVIDIA GPU). pip install -r requirements.txt can't seem to find torch==1.7.1. NVIDIA have pre-built wheels for newer versions of PyTorch including 1.10.0. Will a newer version than 1.7.1 work?

Here's what's available:

https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

znmeb commented 2 years ago

Updates:

  1. The Python on the Jetsons (3.6) is too old for NumPy 1.20.1. Fortunately there's a PPA with newer Pythons and it provides Python 3.9 (https://tecadmin.net/how-to-install-python-3-9-on-ubuntu-18-04/)
sudo add-apt-repository ppa:deadsnakes/ppa

That satisfies all the requirements except torch==1.7.1.

  1. I almost have this installed on my desktop (Windows 11 x86_64, RTX 3090 GPU, WSL Ubuntu 20.04 LTS). torch==1.7.1 doesn't support that GPU but everything else works.

My hope is that everything will work with the latest version of PyTorch, 1.10.0. I don't know yet which is the earliest PyTorch that supports the 3090.

znmeb commented 2 years ago

Latest update: It doesn't look like any of the PyPI arm64 wheels have CUDA support compiled in, and the NVIDIA pre-built wheels for Jetsons only work with Python 3.6. So my plan is to see if the application works with a GPU on x86_64 with more recent versions of PyTorch and then build that PyTorch in-place from source on the Jetson. The instructions for building from source are in https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048 and I'm pretty sure they'll work.

I might even have this working by tomorrow. :-)

znmeb commented 2 years ago

Latest update: install with latest stable GPU-capable PyTorch is working on Windows 11 / WSL / Ubuntu 20.04 LTS. I don't have any tests yet to see if the application works though. https://github.com/AlgoCompSynth/neural-waveshaping-synthesis/blob/win11-wsl-install/win11-wsl-install/README.md.

This is the most powerful GPU I have, so I might attempt training on the URMP dataset.

znmeb commented 2 years ago

Some progress on the Jetson front: https://github.com/AlgoCompSynth/neural-waveshaping-synthesis/issues/1