The right Pytorch for GeForce 3090 & CUDA 11.1

[tyrgann]

Hello,

I’m having troubles installing PyTorch for the workstation we got. I understand that the full power of the GeForce 3090 needs CUDA 11.1 to unveil; are there nightly version I can get with pip or should I try to build from sources, and if yes, from which branch ?

Thanks.

[RaLo4]

This might help:

• Information on building from source with CUDA11

• Information on installing nightly version with CUDA11

[mikey_t]

I faced the same problems with the new 3090 card and cuda 11.1 trying to install pytorch. I followed the directions here to compile pytorch from source. I was able to do this within a conda environment also.