pytorch
2a37f3fd - Relax CUDA architecture check (#45130)

Commit
4 years ago
Relax CUDA architecture check (#45130) Summary: NVIDIA GPUs are binary compatible within major compute capability revision This would prevent: "GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation." messages from appearing, since CUDA-11 do not support code generation for sm_85. Pull Request resolved: https://github.com/pytorch/pytorch/pull/45130 Reviewed By: ngimel Differential Revision: D23841556 Pulled By: malfet fbshipit-source-id: bcfc9e8da63dfe62cdec06909b6c049aaed6a18a
Author
Parents
Loading