[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程

Posted 文火冰糖的硅基工坊

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程相关的知识,希望对你有一定的参考价值。

作者主页(文火冰糖的硅基工坊):https://blog.csdn.net/HiWangWenBing

本文网址:https://blog.csdn.net/HiWangWenBing/article/details/119160541


目录

1. PyTroch概述

2. PyTroch深度学习框架的详细安装过程

2.1 登录到PyTroch官网

2.2 不带GPU硬件平台的PyTroch的安装

2.3 带GPU硬件平台的PyTroch的安装(需要大致10min)

2.4 检验安装


1. PyTroch概述

PyTorch是Facebook推出的一个开源的Python深度学习框架。

PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。

2. PyTroch深度学习框架的详细安装过程

2.1 登录到PyTroch官网

https://pytorch.org/

根据自己的实际环境配置PyTroch,获取安装命令。

2.2 不带GPU硬件平台的PyTroch的安装

备注:选择此安装,在人工神经网络训练时,就不能选择GPU, 只能选择cpu,对于没有GPU的硬件平台,这样的安装可以节省硬盘空间。

conda install pytorch torchvision torchaudio cpuonly -c pytorch

2.3 带GPU硬件平台的PyTroch的安装(需要大致10min)

备注:选择此安装,在人工神经网络训练时,可以选择cpu,也可选择GPU,对于有GPU的硬件平台,这样的选择后续根据的灵活,但需要更多的硬盘工具存放安装包。

激活pytorch环境:conda activate pytorch1.8_py3.8

安装pytorch最新版本: conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

  安装pytorch1.8版本:   conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cpuonly -c pytorch

2.4 检验安装

  • 进入python解析器

(pytorch1.8_py3.8) PS C:\\Users\\Administrator> python                                                                    Python 3.8.10 (default, May 19 2021, 13:12:57) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32

Type "help", "copyright", "credits" or "license" for more information.

  • 导入pytorch库

>>> import torch

  • 调用pytorch库函数,检查GPU是否支持。

>>> torch.cuda.is_available()
False

  • 退出python解析器,回到conda环境

>>> quit()


作者主页(文火冰糖的硅基工坊):https://blog.csdn.net/HiWangWenBing

本文网址:https://blog.csdn.net/HiWangWenBing/article/details/119160541

以上是关于[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程的主要内容,如果未能解决你的问题,请参考以下文章

Pytroch深度学习:从入门到实践

[PyTroch系列-2]:Facebook PyTroch简介生态环境开发架构软件架构

Pytroch同一个优化器优化多个模型的参数并且保存优化后的参数

[PyTroch系列-16]:PyTorch基础 - 张量的操作 - 变形

[PyTroch系列-13]:PyTorch基础 - 张量的操作 - 筛选过滤

[PyTroch系列-11]:PyTorch基础 - 张量Tensor元素的排序