Cannot run pytorch test on GPU

wkcn / AttentionSampler

Attention-based sampler in TASN (Trilinear Attention Sampling Network)

23 stars 0 forks source link

Closed Hhhhhhao closed 2 years ago

Hhhhhhao commented 2 years ago

Hello, thanks for the amazing work on Mobula and AttentionSampler.

I can successfully run the pytorch test file on cpu. But once I change the device to cuda, I get

sh: 1: nvcc: not found

mobula.func.map_step

Do you know what caused this issue? Is it some version issue regarding my cuda? Or nvcc environment path is not seen to Mobula?

Hhhhhhao commented 2 years ago

solved by export cuda into PATH