pytorch学习笔记（一）：pytorch介绍以及安装过程(Windows系统）

原创已于 2023-03-21 09:42:36 修改 · 1.7k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #机器学习 #人工智能

于 2022-04-05 13:41:47 首次发布

Pytorch学习专栏收录该内容

2 篇文章

订阅专栏

本文介绍了PyTorch，一个由Facebook开发的深度学习库，强调其简洁优雅和高效性。学习资源包括Awesome-pytorch-list、官方文档、PyTorch Handbook以及社区交流。教程详细讲解了安装步骤，如创建虚拟环境、换源、GPU配置和安装PyTorch的指导。

部署运行你感兴趣的模型镜像

Pytorch介绍

PyTorch是由Facebook人工智能研究小组开发的一种基于Lua编写的Torch库的Python实现的深度学习库，目前被广泛应用于学术界和工业界，而随着Caffe2项目并入Pytorch， Pytorch开始影响到TensorFlow在深度学习应用框架领域的地位。总的来说，PyTorch是当前难得的简洁优雅且高效快速的框架。

关于Pytorch的学习资料

Awesome-pytorch-list：目前已获12K Star，包含了NLP,CV,常见库，论文实现以及Pytorch的其他项目。
PyTorch官方文档：官方发布的文档，十分丰富。
Pytorch-handbook：GitHub上已经收获14.8K，pytorch手中书。
PyTorch官方社区：在这里你可以和开发pytorch的人们进行交流。

Pytorch的安装

第一：准备条件
首先确保你的电脑有Python环境或者是安装了Anaconda或者Miniconda（个人推荐后面的Anaconda和Miniconda）。
第二：创建虚拟环境

Linux在终端(Ctrl+Alt+T)进行，Windows在Anaconda Prompt进行

查看已经安装好的虚拟环境 可以看到我的电脑已经有几个虚拟环境存在了

conda env list

在这里插入图片描述

创建虚拟环境pytorch

conda create -n pytorch python=python版本名称

虚拟环境常用命令

# 删除虚拟环境命令
conda remove -n 名称 -all

# 激活环境命令
conda activate 名称

第三：Anaconda换源

这部分内容网上很多，建议大家自己查查看。

第四：查看自己的显卡
在cmd中输入nvidia-smi

在这里插入图片描述
看下版本号，看自己可以兼容的CUDA版本，等会安装Pytorch时是可以向下兼容的

在这里插入图片描述
第五：安装Pytorch

Step 1： 进入Pytorch官网**
Step 2：点击Install，进入选择界面
**Step 3：**选择自己电脑可以支持的版本进行下载，复制对应的命令到命令行中下载即可。
**PS： ** conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
使用上述命令时，可以先多试几次，如果一直因为网络原因，下载失败可以尝试
conda install pytorch torchvision torchaudio cudatoolkit=11.3 ，去掉了-c pytorch，conda就会默认从conda源下载，而不是在pytorch官网下载。但是我尝试的过程中，使用清华源下载，经常会默认下cpu版本，而无法下gpu版本的。所以大家多多尝试一下。