Hi, when I try to use pMTnet a waring appears looks like this： <code class="notran

Hi <a class="user-mention notranslate" data-hovercard-type="user" data-hovercard-url="

<a target="_blank" rel="noopener noreferrer nofollow" href="https://user-images.github

Layer lstm_2 will not use cuDNN kernels since it doesn't meet the criteria. about pmtnet HOT 4 OPEN

Whale-fall commented on September 4, 2024

Layer lstm_2 will not use cuDNN kernels since it doesn't meet the criteria.

from pmtnet.

Comments (4)

Whale-fall commented on September 4, 2024

sorry for my poor English, Do you need more info?

from pmtnet.

tianshilu commented on September 4, 2024

Hi @Whale-fall,

Thanks for your interest in our study. This problem is related to the computing environment. The model should still run despite the warnings. If you would like to use cuDNN for acceleration, please refer to the following post. https://www.tensorflow.org/api_docs/python/tf/keras/layers/LSTM.

Thanks!

from pmtnet.

Whale-fall commented on September 4, 2024

Hi!
I am sorry to reply so late. I understand if cuDNN is unavailable.However,I noticed that when I run pMTnet, GPU computing power is basically unused.When tested on a file containing 1000 rows, It spends about 3mins!

My code looks like this:

time python /home/yiliang/software/pMTnet/pMTnet.py \
        -input test.input \
        -library /home/yiliang/software/pMTnet/library \
        -output ./ -output_log ./test_log &

However, from my perspective, It is conflicts with the performance claimed in this issue. It says pMTnet can finish prediction in 2mins on a file containing 2M rows. Can you help me fix this?

This is My env info：

keras                     2.4.3              pyhd8ed1ab_0    conda-forge
keras-preprocessing       1.1.2              pyhd8ed1ab_0    conda-forge
numpy                     1.19.5           py37h3e96413_3    conda-forge
pandas                    1.3.5            py37he8f5f7f_0    conda-forge
python                    3.7.12          hf930737_100_cpython    conda-forge
python-dateutil           2.8.2              pyhd8ed1ab_0    conda-forge
python-flatbuffers        2.0                pyhd8ed1ab_0    conda-forge
python_abi                3.7                     2_cp37m    conda-forge
scikit-learn              1.0.2            py37hf9e9bfc_0    conda-forge
scipy                     1.7.3            py37h492baa0_0    defaults
tensorflow                2.4.1           gpu_py37ha2e99fa_0    defaults
tensorflow-base           2.4.1           gpu_py37h29c2da4_0    defaults
tensorflow-estimator      2.6.0            py37hcd2ae1e_0    conda-forge
tensorflow-gpu            2.4.1                h30adc30_0    defaults

from pmtnet.

Whale-fall commented on September 4, 2024

@tianshilu @Akazhiel