【ML】windows 安装使用pytorch

使用pytorch需要python环境，建议是直接装anaconda ，IDE用visual studio

anaconda安装

Anaconda 是一个用于科学计算的 Python 发行版，支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包
官网链接anaconda
本人下载：Anaconda3-2023.03-1-Windows-x86_64.exe 内置python 3.10
双击exe一键安装，安装好后可以通过anaconda navigator查看已安装的包

可以通过 Start - Anaconda3 - Anaconda PowerShell Prompt查看安装版本

pytorch 安装

官网
quick start guide

按需选择配置获取安装cmd，推荐windows下配置如下
PyTorch build – stable.
Your OS – Windows
Package – Conda
Language – Python
Compute Platform – CPU, 或者根据你Cuda版本选择合适的compute platform
cuda版本可以点nvdia系统配置从driver处看，比如下图第三行显示是11.6

也可以命令行运行nvdia-smi获取信息。

在 Anaconda prompt 命令窗口中输入cmd 开始安装 eg： conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

安装结束确认

pytorch 使用

以图片分类demo为例使用pytorch

1. visual studio 创建新project
   

2. 点Create a new project后选择Python Application，新建python项目，（直接运行项目等上传后贴链接 点击这里）

3. 配置环境，添加环境选现有环境，如果默认没出来anaconda，就自定义环境，把anaconda所在路径添加进去，vs会自动识别，确认添加即可
   

4. demo正文开始，先load 数据集
   load CIFAR10, 第一次加载需要比较多的时间

from torchvision.datasets import CIFAR10
from torchvision.transforms impor