weixin_48298208的博客

原创 Pytorch卷积神经网络GPU加速

Pytorch卷积神经网络GPU加速首先要配置好pytorch，通常我们直接pip install pytorch的话会缺少关于cude的内容，可以通过torch.cuda.is_available()来判断是否需要重新安装pytorch如果返回False 我们需要到官方重新下载pytorch（https://pytorch.org/），在此之前我们可以pip uninstall 来卸载已有的torch包。1.官方找到对应自己电脑版本的包首先查看自己电脑cude版本，打开英伟达控制面板—帮助—系统