PyTorch的安装

PyTorch的安装

  • 1.什么是Pytorch,为什么选择Pytroch?
  • 2.PyTorch的安装
  • 3.PyTorch基础概念

1.什么是Pytorch,为什么选择Pytroch?

PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序。 它主要由Facebook的人工智能研究小组开发。Uber的"Pyro"也是使用的这个库。

PyTorch是一个Python包,提供两个高级功能:

  1. 具有强大的GPU加速的张量计算(如NumPy)
  2. 包含自动求导系统的的深度神经网络

2.PyTorch的安装

先用anaconda的navigator直接搜索PyTorch安装,装好了之后测试,结果出现问题
PyTorch的安装_第1张图片
PyTorch的安装_第2张图片
CUDA的版本不一致,然后卸载,重装
conda create -n pyTorch python=3.6
conda active pyTorch
然后运行官网上的命令
PyTorch的安装_第3张图片
安装成功,测试代码

# CUDA TEST
import torch
x = torch.Tensor([1.0])
xx = x.cuda()
print(xx)

# CUDNN TEST
from torch.backends import cudnn
print(cudnn.is_acceptable(xx))

输出结果:
在这里插入图片描述

3.PyTorch基础概念

import torch
# 创建一个 5x3 矩阵, 但是未初始化:
x = torch.empty(5, 3)
print(x)
# 创建一个随机初始化的矩阵:
x = torch.rand(5, 3)
print(x)
# 创建一个0填充的矩阵,数据类型为long:
x = torch.zeros(5, 3, dtype=torch.long)
print(x)
# 创建tensor并使用现有数据初始化:
x = torch.tensor([5.5, 3])
print(x)
# 根据现有的张量创建张量。 这些方法将重用输入张量的属性,例如, dtype,除非设置新的值进行覆盖
x = x.new_ones(5, 3, dtype=torch.double)      # new_* 方法来创建对象
print(x)
x = torch.randn_like(x, dtype=torch.float)    # 覆盖 dtype!
print(x)

PyTorch的安装_第4张图片
PyTorch的安装_第5张图片

你可能感兴趣的:(PyTorch)