Thank you for your reply.
For the CUDA version here is what I have from nvcc:
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 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 …
