RTX 3000 Support

1

Hello, I’m getting this error whenever I try to run anything with Torch:

UserWarning: 
GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_61 sm_70 sm_75 compute_37.
If you want to use the GeForce RTX 3090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

Does this simply mean that 3090’s aren’t supported yet or is there something I can do to enable support. Currently using CUDA 10.2, PyTorch 1.6

Thanks in advance

TeraChad

2 Likes
2

you may want to try it with cuda 11:
pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cu110/torch_nightly.html

7 Likes
3

That seems to have fixed it, thank you

4

Are you solve your problem? I will buy a 3090 and i hesitate to buy it!!!

5

Are you solve your problem? I will buy a 3090 and i hesitate to buy it!!!

You don’t get to use the full power of RTX-30*, but otherwise it works just fine with pytorch/cuda-11.0