gpu版本pytorch配置

pytorch官网可以下载

https://pytorch.org/

先打开命令行窗口（windows+r），输入nvidia-smi查看电脑能支持的最高cuda版本

进入pytorch官网，我选择下载官网的cuda10.2版本对应的pytorch,所以点击下图以前的版本

我刚开始直接安装的最新版本pytorch（设置了清华镜像源网站），但是网站有时候会推送cpu版本的pytorch，在配置环境的时候，注意查看推送的版本，比如先创建环境，然后在pytorch环境中下载pytorch,使用上图的命令

conda create --name pytorch python=3.7

然后发现推送的是cpu版本（我这里是因为已经安装了pytorch