配置pytorch GPU流程

一、安装cuda和cuDNN

vs其实是可以不用安装的

CUDA安装及环境配置——最新详细版-CSDN博客

保证上面的cuda安装无误后

在python环境里输入

import torch
torch.cuda.is_available()

ok，是false，说明还是有问题，pytorch的事情

二、安装正确的pytorch

打开你的codna虚拟环境运行，卸载当前的torch版本

pip uninstall torch torchvision torchaudio

pytorch下载链接https://pytorch.org/get-started/locally/找到你下载的对应的版本然后在anaconda虚拟环境种配置即可

接着就是安装啊吧啊吧一大堆

安装完成后，进入python，输入

import torch
torch.cuda.is_available()

ok，返回了true，可以用了加速深度学习了！