Pytorch-RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment

刚装好GPU版本的Pytorch报了这个错，仔细核对了一下Pytorch的版本和我的cuda版本，是一致的。nvidia-smi运行没有问题
错误如下：

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.

然后 我按照它给的方式执行了一下

export CUDA_VISIBLE_DEVICES='0,1,2,3

结果还是报这个错，最后发现原来是更新了drivers没有重启，
最后reboot了一下，解决！