[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程
Posted 文火冰糖的硅基工坊
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程相关的知识,希望对你有一定的参考价值。
作者主页(文火冰糖的硅基工坊):https://blog.csdn.net/HiWangWenBing
本文网址:https://blog.csdn.net/HiWangWenBing/article/details/119160541
目录
2.3 带GPU硬件平台的PyTroch的安装(需要大致10min)
1. PyTroch概述
PyTorch是Facebook推出的一个开源的Python深度学习框架。
PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。
2. PyTroch深度学习框架的详细安装过程
2.1 登录到PyTroch官网
根据自己的实际环境配置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系列-2]:Facebook PyTroch简介生态环境开发架构软件架构
Pytroch同一个优化器优化多个模型的参数并且保存优化后的参数
[PyTroch系列-16]:PyTorch基础 - 张量的操作 - 变形