之前不好好裝CUDA,在執行的時候報了一個類似torch._C._cuda_init() RuntimeError: CUDA error: unknown error 的錯誤
cuda不好好裝引起的。
去nvidia官網再找cudatoolkit裝一遍就行了。
最新版可能和系統不相容,這個時候把驅動全部刪掉再裝一遍就行。
之前不好好裝CUDA,在執行的時候報了一個類似torch._C._cuda_init() RuntimeError: CUDA error: unknown error 的錯誤
cuda不好好裝引起的。
去nvidia官網再找cudatoolkit裝一遍就行了。
最新版可能和系統不相容,這個時候把驅動全部刪掉再裝一遍就行。