PyTorch安装
一、Anaconda安装
二、Cudnn和CUDA的安装
查看本机的CUDA版本(可在cmd中输入nvidia-smi 来查看)，并下载与之匹配的cudnn
三、pytorch安装
打开anaconda prompt，在命令提示符中输入
# 创建虚拟环境名为pytorch
conda create –n pytorch python=3.6
# 激活环境
conda activate pytorch
# 安装pytorch
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html