pytorch安装

1、Anaconda的安装（自行参考其他博客）
2、使用pip安装pytorch

• 打开anaconda prompt创建一个pytorch的环境
• conda create -n pytorch_gpu python=3.7
• pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
• 安装COUD11(可以查看自己显卡的种类去pytorch官网上指定下载)、
• 测试
• import
• torch.cuda.is_available()
• end
  **注意：**在安装的过程中建议不要安装在C盘，在后期需要训练模型的时候需要下载各种安装包，容易造成C盘满载

参考文献