600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > [PyTroch系列-1]:PyTroch深度学习框架的详细安装过程

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

时间:2019-09-02 02:06:18

相关推荐

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

作者主页(文火冰糖的硅基工坊):文火冰糖(王文兵)的博客_文火冰糖的硅基工坊_CSDN博客

本文网址:[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程_文火冰糖(王文兵)的博客-CSDN博客

目录

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官网

PyTorch

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

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

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

conda install pytorch torchvision torchaudio cpuonly -c pytorch

2.3 带GPU硬件平台的PyTroch的安装(需要大致1小时)

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

创建pytorch环境:conda create -npytorch1.8_py3.8

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

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

安装pytorch 指定版本:conda install pytorch==1.8.0torchvision==0.9.0 torchaudio=0.8.0cudatoolkit=11.1-c pytorch

2.4 检验安装

进入python解析器

(pytorch1.8_py3.8) PS C:\Users\Administrator>pythonPython 3.8.10 (default, May 19 , 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()

第3章 安装其他的常用库

conda install matplotlib

作者主页(文火冰糖的硅基工坊):文火冰糖(王文兵)的博客_文火冰糖的硅基工坊_CSDN博客

本文网址:[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程_文火冰糖(王文兵)的博客-CSDN博客

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。