問題原因

訓練太慢大多是因為沒有安裝CUDA和pytorch，導致的只有cpu在跑，顯卡沒跑

這就是很典型的。

解決方案

第一步：安裝CUDA

在本機上面安裝CUDA,記住只有N卡可以安裝，一開始的電腦是自帶CUDA的。

如果不是自帶的CUDA，下載落后一點的，可能打游戲的時候會有點毛病。

nvcc -V

本機安裝成功后，其他虛擬環境也是有的。

至于安裝CUDA的教程，我就不教了，網上一大片都有。

第二步：安裝pytorch

官網：Previous PyTorch Versions | PyTorch

根據CUDA的版本挑選合適的。

?就比如說我這個，是10.2 版本的。

就這個了。