Thank you for your reply.
For the CUDA version here is what I have from nvcc:
nvcc: NVIDIA ® Cuda compiler driver
Copyright © 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_13:16:23_CDT_2017
Cuda compilation tools, release 9.0, V9.0.175
When I try running:
export TORCH_CUDA_ARCH_LIST=6.1
python3 setup.py install
Then I get the same errors. I attach the full output screenshot if that helps …

Capture d’écran 2019-09-13 à 13.07.50.png950×1142 181 KB