【PyTorch】环境配置

发布于:2024-10-10 ⋅ 阅读:(9) ⋅ 点赞:(0)

框架介绍

Pytorch简介

2017年1月,FAIR(Facebook AI Research)发布了PyTorch。PyTorch是在Torch基础上用python语言重新打造的一款深度学习框架。Torch是采用Lua语言作为接口的机器学习框架,但因为Lua语言较为小众,导致Torch知名度不高。

PyTorch发展

  • 2017年1月正式发布PyTorch
  • 2018年4月更新0.4.0版,支持Windows系统,caffe2正式并入PyTorch
  • 2018年11月更新1.0稳定版,已GitHub 增长第二快的开源项目
  • 2019年5月更新1.1.0版,支持TensorBoard,增强可视化功能
  • 2019年8月更新1.2.0版,更新torchvision,torchaudio 和torchtext,增加更多功能

PyTorch优点

  • 上手快:掌握Numpy和基本深度学习概率即可上手
  • 代码简单灵活:用nn.module封装使得网络搭建更加方便;基于动态图机制,更灵活
  • Debug方便:调试PyTorch就像调试Python代码一样简单
  • 文档规范:https://pytorch.org/docs/stable/index.html 可查看各版本文档
  • 资源多:arXiv中的新算法大多有PyTorch实现
  • 开发者多:GitHub上贡献者已经超过1100+
  • 背靠大树:FaceBook维护开发
  • ……

适合人群

  • 深度学习初学者:模型算法实现容易,加深深度学习概念认识
  • 机器学习爱好者:数十行代码便可实现人脸识别,目标检测,图像生成等有趣实验
  • 算法研究员:最新arXiv论文算法快速复现

软件安装

Python包管理器

Anaconda是为了方便使用python而建立的一个软件包,其中包含常用的250多个工具包,多版本python解释器和强大的虚拟环境管理,所以Anaconda得名python全家桶。

Anaconda可以使安装、运行和升级环境变得更简单,因此推荐安装使用。
在这里插入图片描述
安装步骤:

  1. 官网下载安装包
  2. 运行安装包
  3. 选择路径,勾选添加环境变量,等待安装完成
  4. 严重安装成功,打开cmd,输入conda,回车
  5. 添加中科大镜像或者清华镜像源

Python集成开发环境

PyCharm —— 强大的python IDE,拥有调试语法高亮、Project管理、代码跳转智能提示、版本控制等功能。
在这里插入图片描述
安装步骤:

  1. 官网下载安装包
  2. 运行安装包
  3. 选择路径,勾选添加环境变量,等待安装完成

PyTorch

在这里插入图片描述
安装步骤:
1、检测是否有合适的GPU,若有,需要安装CUDACuDNN
2、CUDACuDNN安装(非必须)
3、官网寻找对应版本下载(cuda版本或cpu、pytorch版本、python版本、操作系统都需要一一对应)

CUDA安装(补充)

CUDA(Compute Unified Device Architecture),是一种由NVIDIA推出的通用并行计算架构
CuDNN:为深度学习计算设计的软件库

安装步骤:

  1. 检查pytorch版本所支持的cuda版本,下载对应版本的CUDA
  2. 下载CUDA安装包,下载对应CUDA版本的CuDNN安装包
  3. 安装并验证安装是否成功

解释器与工具包

在这里插入图片描述
在这里插入图片描述

虚拟环境 Vittual Environment

在这里插入图片描述

Window11+RTX4060配置CUDA11.8+pytorch2.0.0

下载CUDA11.8

进入官网下载Link,然后点击Archive of Previous CUDA Releases,找到对应版本11.8。
在这里插入图片描述
在这里插入图片描述

选择对应操作系统版本,点Download下载到本地。
在这里插入图片描述

下载cuDNN

点击链接Link,找到CUDA11.8对应的cuDNN。在这里插入图片描述

安装CUDA11.8

全程点击next直到安装结束。
在这里插入图片描述
配置环境变量,在path里添加。

#前两个应该自动配好了
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\include
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\lib

在这里插入图片描述
在命令行窗口输入以下命令看是否成功

nvcc -V

成功显示以下内容表示安装成功
在这里插入图片描述

安装cuDNN

解压下载的压缩包
在这里插入图片描述
把里边的三个文件夹bin,include,lib复制到下面路径(如果改了CUDA的安装路径,就复制到对应文件夹里)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

在这里插入图片描述

下载并安装python

官网链接
在这里插入图片描述

下载并安装Anaconda

官网链接选择适合自己系统的 64 位安装包,注意选择 Python3 以上的版本。

安装torch

进入官网Link,找到v2.0.0,用pip语句安装。用conda可能会出错,具体见安装pytorch报错torch.cuda.is_available()=false的解决方法

# CUDA 11.8
pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

在这里插入图片描述

参考链接

win11+4060配置CUDA11.8+pytorch2.0.0