之前不好好装CUDA,在运行的时候报了一个类似torch._C._cuda_init() RuntimeError: CUDA error: unknown error 的错误
cuda不好好装引起的。
去nvidia官网再找cudatoolkit装一遍就行了。
最新版可能和系统不兼容,这个时候把驱动全部删掉再装一遍就行。