v1.8.0-torchml-beta1
Pre-release
Pre-release
·
28 commits
to 26-torchml
since this release
4aaa14a
This commit was signed with the committer’s verified signature.
ilia-nikiforov-umn
ilia Nikiforov
Set TORCH_CUDA_ARCH_LIST=Common in the environment by default. User can override with specific architecture. Auto-detecting architecture doesn't work, the auto-detection mechanism in the Caffe2 makefiles in libtorch causes a "CUDA98" dialect error.