Damon Da Tong
Damon Da Tong
same with me. versions: python: 3.10.14 decord: 0.6.0 pytorch: 2.3.0 and cuda related libs installed by pip: nvidia-cublas-cu12 12.1.3.1 nvidia-cuda-cupti-cu12 12.1.105 nvidia-cuda-nvrtc-cu12 12.1.105 nvidia-cuda-runtime-cu12 12.1.105 nvidia-cudnn-cu12 8.9.2.26 nvidia-cufft-cu12 11.0.2.54 nvidia-curand-cu12...
Found something helpful: `tf.contrib.layers.scattered_embedding_lookup`. reference: http://arxiv.org/pdf/1504.04788.pdf
I got a problem when compiling the roi_pooling_op. Since I do not have GPU installed, I have configured bazel using cpu only. But when I tried to load the lib...
Yes, I understand. I have just commented lines related to GPU code. Thanks.