问题原因

训练太慢大多是因为没有安装CUDA和pytorch，导致的只有cpu在跑，显卡没跑

这就是很典型的。

解决方案

第一步：安装CUDA

在本机上面安装CUDA,记住只有N卡可以安装，一开始的电脑是自带CUDA的。

如果不是自带的CUDA，下载落后一点的，可能打游戏的时候会有点毛病。

nvcc -V

本机安装成功后，其他虚拟环境也是有的。

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

第二步：安装pytorch

官网：Previous PyTorch Versions | PyTorch

根据CUDA的版本挑选合适的。

 就比如说我这个，是10.2 版本的。

就这个了。