Hi,
If you want pytorch, you don’t need to compile full caffe2, is there a reason you use the FULL_CAFFE2 flag? I’m not even sure it exists any more.
You can find the the setup.py here 86 all the env variables used to find MKLDNN.
The cuda architecture list should be detected automatically, you don’t need to specify it by hand.