search

sniklaus / 3d-ken-burns

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch
Other
1.52k stars 225 forks source link

pytorch version and cupy version? #15

Closed LonglongaaaGo closed 4 years ago

LonglongaaaGo commented 4 years ago

Hello! Thanks for your repo！ It is a very beautiful project ! Recently, I want to try your code, and I tried to run on GeForce RTX 2080 with pytorch 1.3.1, cuda 10.0 and K40 with pytorch 1.3.1, cuda 9.2 10.0 in different machine, respectively, but I get the approximately the same error as fllow: "RuntimeError: CUDA error: no kernel image is available for execution on the device (nms_cuda at /tmp/pip-req-build-c2_g4c3l/torchvision/csrc/cuda/nms_cuda.cu:127)" so , can you give me some advise for running this project? Thank you so much!

LonglongaaaGo commented 4 years ago

one machine use the environment with GeForce RTX 2080 with pytorch 1.3.1, cuda 10.0 and another with K40 , pytorch 1.3.1 and cuda 9.2

sniklaus commented 4 years ago

It seems like the error is not related to the provided implementation. The discussion here indicates that your TorchVision is not compatible with your PyTorch: https://github.com/facebookresearch/detectron2/issues/469

sniklaus commented 4 years ago

Closing due to inactivity, feel free to reopen in case the issue prevails and is related to this project.