CUDA error

Run the following to fix the error:

``` sudo rmmod nvidia_uvm sudo modprobe nvidia_uvm ```

Error: ``` RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. ```