The right Pytorch for GeForce 3090 & CUDA 11.1

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.

2 Likes

This might help:

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.