peizhaoli05

I am having the same error while using PyTorch 1.8.1/1.7.1.

My environment: driver 460.32.03, cuda 11.1 (by nvcc -V). In my case: Both PyTorch 1.8.1 and 1.7.1 and compile https://github.com/jinfagang/DCNv2_latest and cannot pass the _testcuda.py_ without changing the tolerance. Only...

Hi @andrewsilva9 , Thanks for your reply. Actually I assume that put `h_init` into the function `hvp` is not right for calculating `hv` according to the paper. When I change...

I encounter the same issue. I am using CUDA 11.1 with Jax 0.2.18. ``` 2021-08-11 22:19:59.368992: W external/org_tensorflow/tensorflow/stream_executor/gpu/asm_compiler.cc:235] Falling back to the CUDA driver for PTX compilation; ptxas does not...

Hi @hawkinsp I think I'm using a Cuda version that is supported by JAX. Do you know how to find the ptxas path found by JAX? Thanks.