Cuda error when using GPU

Hello, I use torch 1.9.0 and uda 10.2 , GeForce RTX3090 but I got this error:

You would need to install the CUDA11.1 binary to use your Ampere GPU.

1 Like