英伟达驱动问题：
AssertionError: CUDA unavailable, invalid device 0 requested
$ nvidia-smi
NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running.

解决方法：显卡驱动报错：NVIDIA-SMI has failed because it couldn’t communicate with the NVIDIA driver.