Nvcc fatal: Unsupported gpu architecture 'compute_70'

Hi everybody,

our sever has GeForce TitanX and a Pscal V100 GPU, when i compile the pytorch v0.4.0, it throw the error above, how could i fix it? The error is encountered when compiling the ATen.


The problem is that your cuda install is too old and does not know about the Volta card.
You would need to upgrade cuda.

I install the cuda8.0 and it work fine when install from pip, while i want install from source for some reason.

You will need cuda 9.1 to be able to compile I think.

OK, thanks for your reply. It may be. And I found a solution which do not know whether it may cause other problem. I set the enviroment variable TORCH_CUDA_ARCH_LIST=6.0 and install the pytorch.

If you do so, you will be able to use the TitanX card but not the V100 one.

I can use both, but it will give warning