【ML】windows 安装使用pytorch_window pytorch

作者：人工智能uu | 2024-08-12 03:37:31

踩

window 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查看安装版本
check

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

visual studio 创建新project
点Create a new project后选择Python Application，新建python项目，（直接运行项目等上传后贴链接点击这里）
配置环境，添加环境选现有环境，如果默认没出来anaconda，就自定义环境，把anaconda所在路径添加进去，vs会自动识别，确认添加即可
demo正文开始，先load 数据集
load CIFAR10, 第一次加载需要比较多的时间

from torchvision.datasets import CIFAR10
from torchvision.transforms impor1

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/人工智能uu/article/detail/967495