Error executing pycuda code: "nvcc fatal: Value 'sm_61' is not defined for option 'gpu-architecture'

I just installed pycuda on Ubuntu 16.04 and am trying to verify that it is working. I'm trying to run:

#https://documen.tician.de/pycuda/tutorial.html

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule


import numpy
a = numpy.random.randn(4,4)


a = a.astype(numpy.float32)

a_gpu = cuda.mem_alloc(a.nbytes)

cuda.memcpy_htod(a_gpu, a)


mod = SourceModule("""
  __global__ void doublify(float *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)

Error:

pycuda.driver.CompileError: nvcc compilation of /tmp/tmpzohid21m/kernel.cu failed
[command: nvcc --cubin -arch sm_61 -I/usr/local/lib/python3.5/dist-packages/pycuda/cuda kernel.cu]
[stderr:
nvcc fatal   : Value 'sm_61' is not defined for option 'gpu-architecture'
]

Any ideas how to solve it?