search

charlesq34 / pointnet

PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation
Other
4.73k stars 1.45k forks source link

cudnn version mismatch #82

Closed amiltonwong closed 6 years ago

amiltonwong commented 6 years ago

Hi, authors,

I got the following error when I run python train.py under part_seg directory. It seems my current cudnn (v6) is newer than your provided code. Any workaround steps to pass these issues?

THX!

I tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA: 0 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0:   Y 
I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: TITAN Xp, pci bus id: 0000:01:00.0)

<<< Testing on the test dataset ...
Loading test file /data/code3/pointnet/part_seg/./hdf5_data/ply_data_val0.h5
E tensorflow/stream_executor/cuda/cuda_dnn.cc:390] Loaded runtime CuDNN library: 6021 (compatibility version 6000) but source was compiled with 5110 (compatibility version 5100).  If using a binary install, upgrade your CuDNN library to match.  If building from sources, make sure the library loaded at runtime matches a compatible version specified during compile configuration.
F tensorflow/core/kernels/conv_ops.cc:605] Check failed: stream->parent()->GetConvolveAlgorithms(&algorithms) 
Aborted (core dumped)
charlesq34 commented 6 years ago

Hi @amiltonwong

It seems to me a TensorFlow installation problem. Have you successfully run the classification script or other tensorflow example code?

Best, Charles

amiltonwong commented 6 years ago

I' ll test it further...

amiltonwong commented 6 years ago

Yes, Charles, You're right, after reinstalling cudnn version 5, the training process works under part_seg directory.