LearnablePromptSAM cuDNN error

which version of pytorch is used, could you share the package lists?

Sep 06 '23 10:09 liuyanice

Good issue, we use torch 1.13.1+CU117. It seems that the model or data can not be feed into the GPU. I suggest you check the version of CUDA.

Sep 06 '23 11:09 Qsingle

my issue： torch 1.13.1+CU117 nvidia-smi: CUDA Version:11.7 Driver Version:515.105.01 nvcc: cuda_11.3.r11.3

cuDNN error: CUDNN_STATUS_NOT_INITIALIZED

Sep 06 '23 15:09 liuyanice

Can you tell me the GPU you use? The version of cudnn. If you use Anaconda or Minianaconda I suggest you to create one virtual environment by the following command. Or you can install the CUDA 11.7 and Cudnn 8.5.0.

conda create -n pytorch python=3.8 pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
conda activate pytorch #activate the enviroment

Sep 07 '23 03:09 Qsingle

RTX TI3090 single GPU, can you tell me the GPU you use?

Sep 11 '23 03:09 liuyanice

I know the reason for this error. Similar to #6, your label may contain an invalid value like 255.

Sep 11 '23 06:09 Qsingle

Can you share your datasets, I want to reproduce the results.

Sep 11 '23 07:09 liuyanice

Beacause I have set num classes is 2 and used the blood vessel datasets. It also displays "cudnn error".

Sep 11 '23 07:09 liuyanice

Thank you, I have already solved this problem.

Sep 11 '23 07:09 liuyanice

It is indeed caused by an invalid value

Sep 11 '23 07:09 liuyanice