问题原因
训练太慢大多是因为没有安装CUDA和pytorch,导致的只有cpu在跑,显卡没跑
这就是很典型的。
解决方案
第一步:安装CUDA
在本机上面安装CUDA,记住只有N卡可以安装,一开始的电脑是自带CUDA的。
如果不是自带的CUDA,下载落后一点的,可能打游戏的时候会有点毛病。
nvcc -V
本机安装成功后,其他虚拟环境也是有的。
至于安装CUDA的教程,我就不教了,网上一大片都有。
第二步:安装pytorch
官网:Previous PyTorch Versions | PyTorch
根据CUDA的版本挑选合适的。
就比如说我这个,是10.2 版本的。
就这个了。
兄弟们,中计了,这个是默认CPU的
这个才是GPU的,不过网速巨慢