Nvcc fatal: Unsupported gpu architecture 'compute_70'

1

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.

2

Hi,

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

3

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

4

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

5

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.

6

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

7

I can use both, but it will give warning