Describe the Bug
No module named 'amp_C'
Minimal Steps/Code to Reproduce the Bug
use ```python setup.py install --cuda_ext```
if ```RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries. ```you can commenting out this check (```check_cuda_torch_binary_vs_bare_metal(CUDA_HOME)```) in setup.py,178
**Expected Behavior**
Describe the Bug No module named 'amp_C' Minimal Steps/Code to Reproduce the Bug
use ```python setup.py install --cuda_ext``` if ```RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries. ```you can commenting out this check (```check_cuda_torch_binary_vs_bare_metal(CUDA_HOME)```) in setup.py,178 **Expected Behavior**Environment