深度学习：Pytorch（gpu)在windows10上的安装

一、下载anaconda，cuda，cudnn

这三个必须提前下载好，下载教程可以参考深度学习：Windows10+CUDA10.0+CUDNN7.4

二、下载pytorch

有两种方法：

方法一：去pytorch官网按命令行下载，不过比较慢，不推荐，传送门

1、根据自己的电脑（windows，Linux，mac选择）
2、使用conda安装
3、python编译环境
4、cuda版本号，这个根据自己之前安装的cuda版本号来确定
由于我的cuda：11.3，所以就选择这个，
下载：conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

方法二：把pytorch下载到本地，然后在命令行中下载本地的pytorch

下载链接：https://download.pytorch.org/whl/torch_stable.html
至于下载哪个版本，在下图表中选择：图表来自

打开下载链接后，根据下表选择合适自己版本的pytorch文件包，保存该文件到任意文件夹（此路径得记住，等会会用到）

下载并保存文件夹后，打开Anaconda prompt
1、创建一个虚拟环境：名为pytorch

conda create --name pytorch python=3.6 anaconda

2、激活该虚拟环境

activate pytorch

3、安装torch
找到之前下载的pytorch的绝对路径，E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl，则直接下载：

pip install E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl

4、到此就安装成功
5、判断是不是gpu

#在该虚拟环境下，进入python
import torch
print(torch.cuda.is_available)