英伟达驱动问题:
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.