为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?
为什么有GPU,使用pytorch训练的时候还是使用的CPU训练?前言`提示:因为使用pip进行安装,默认下载的就是cpu的版本:` ![在这里插入图片描述](https://img-blog.csdnimg.cn/1c4dc1d99bb84b1596685e4cd9b7497e.png)解决方法1、进入https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/下载安装包2、选择对应的cuda版本进行下载如果不知道cuda的版本在控制台中输入nvidia-smi
复制链接