GPU version finally works for me. Here are the steps...
- Dominant language
- Python
- Stars
- 28.4k
- Forks
- 3.1k
- PR merge metrics
- No merged PRs in 30d
Description
I spent the entire weekend trying to get spleeeter to work with GPU (maybe I'm slow...). I could make combinations of versions work with the CPU, but the command line always failed with the GPU enabled. The following steps finally worked!
I don't show the steps to clone the spleeter project from github, since these files were already in place from previous attempts. Also note: I only tested the command line version of 'separate.' Since this worked, I'm going to assume (for now) that the problem is solved. At least it is running with the GPU. If I had more time, I suppose I could increase the version of cudnn and the tools until it failed. I'd be interested if someone else got it working with a more recent rev.
I have an Asus G14 Zephyrus with an NVIDIA GeForce RTX 3060 Laptop GPU and an AMD Ryzen 9 5900HS.
Here are the steps:
1. From Anaconda Navigator, I created a new environment with Python 3.9.7 and a CMD.exe Prompt.
2. From the new CMD Prompt, I typed the following commands:
(Python_3_9_7) C:\Users\windo>conda create -n tensorflow2.5 python=3.9
(Python_3_9_7) C:\Users\windo>conda activate tensorflow2.5
(tensorflow2.5) C:\Users\windo>conda install cudatoolkit=10.0.130
(tensorflow2.5) C:\Users\windo>conda install cudnn=7.3.1
(tensorflow2.5) C:\Users\windo>pip3 install tensorflow-gpu==2.5
(tensorflow2.5) C:\Users\windo>conda install -c conda-forge ffmpeg libsndfile
(tensorflow2.5) C:\Users\windo>pip install spleeter
(tensorflow2.5) C:\Users\windo>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Thu_Feb_10_19:03:51_Pacific_Standard_Time_2022
Cuda compilation tools, release 11.6, V11.6.112
Build cuda_11.6.r11.6/compiler.30978841_0
**For reference, here is the list of of my environment, including version numbers:**
(tensorflow2.5) C:\Users\windo>conda list
Packages in environment at C:\Anaconda\envs\tensorflow2.5:
Name Version Build Channel
absl-py 0.15.0 pypi_0 pypi
astunparse 1.6.3 pypi_0 pypi
ca-certificates 2022.2.1 haa95532_0
cachetools 5.0.0 pypi_0 pypi
certifi 2021.10.8 py39haa95532_2
charset-normalizer 2.0.12 pypi_0 pypi
cudatoolkit 10.0.130 0
cudnn 7.3.1 cuda10.0_0
flatbuffers 1.12 pypi_0 pypi
gast 0.4.0 pypi_0 pypi
google-auth 2.6.0 pypi_0 pypi
google-auth-oauthlib 0.4.6 pypi_0 pypi
google-pasta 0.2.0 pypi_0 pypi
grpcio 1.34.1 pypi_0 pypi
h5py 3.1.0 pypi_0 pypi
idna 3.3 pypi_0 pypi
importlib-metadata 4.11.3 pypi_0 pypi
keras-nightly 2.5.0.dev2021032900 pypi_0 pypi
keras-preprocessing 1.1.2 pypi_0 pypi
markdown 3.3.6 pypi_0 pypi
numpy 1.19.5 pypi_0 pypi
oauthlib 3.2.0 pypi_0 pypi
openssl 1.1.1m h2bbff1b_0
opt-einsum 3.3.0 pypi_0 pypi
pip 21.2.4 py39haa95532_0
protobuf 3.19.4 pypi_0 pypi
pyasn1 0.4.8 pypi_0 pypi
pyasn1-modules 0.2.8 pypi_0 pypi
python 3.9.7 h6244533_1
requests 2.27.1 pypi_0 pypi
requests-oauthlib 1.3.1 pypi_0 pypi
rsa 4.8 pypi_0 pypi
setuptools 58.0.4 py39haa95532_0
six 1.15.0 pypi_0 pypi
sqlite 3.38.0 h2bbff1b_0
tensorboard 2.8.0 pypi_0 pypi
tensorboard-data-server 0.6.1 pypi_0 pypi
tensorboard-plugin-wit 1.8.1 pypi_0 pypi
tensorflow-estimator 2.5.0 pypi_0 pypi
tensorflow-gpu 2.5.0 pypi_0 pypi
termcolor 1.1.0 pypi_0 pypi
typing-extensions 3.7.4.3 pypi_0 pypi
tzdata 2021e hda174b7_0
urllib3 1.26.8 pypi_0 pypi
vc 14.2 h21ff451_1
vs2015_runtime 14.27.29016 h5e58377_2
werkzeug 2.0.3 pypi_0 pypi
wheel 0.37.1 pyhd3eb1b0_0
wincertstore 0.2 py39haa95532_2
wrapt 1.12.1 pypi_0 pypi
zipp 3.7.0 pypi_0 pypi
Hope this helps someone!
Contributor guide
Assessment
This issue has not been assessed yet.