Relationship between CMake build and python setup.py

deployment
2

I got it… Problem solved…
It looks I can do

  1. CMake build FIRST
  2. TORCH_CUDA_ARCH_LIST="8.6" NO_TEST=1 USE_MKLDNN=0 FULL_CAFFE2=1 python setup.py bdist_wheel
1 Like