PyTorch学习笔记

2020-02-13 04:10:13 64 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

龙良曲pytorch基础课程学习笔记什么是Pytorch? PyTorch 是一个基于 Python 的科学计算包，主要定位两类人群： NumPy 的替代品，可以利用 GPU 的性能进行计算深度学习研究平台拥有足够的灵活性和速度开始学习 tensor对象在 TensorFlow 中，数据不是以整数，浮点数或者字符串形式存在的，而是被封装在一个叫做 tensor 的对象中。Tensor是张量的意思，张量包含了0到任意维度的量... 阅读详情

PyTorch
# 安装NVIDIA cuda
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10

# GPU compute capability <= 3.5 无法使用1.3版本及以上的PyTorch
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

# 清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

# 下载
conda install pytorch torchvision cudatoolkit=10.1

# 检验运行成功
import torch
import torchvision
a = torch.Tensor(5, 3)
a = a.cuda()
print(a)

# 常用包调用
import torch
import torch.nn.functional as F
import torch.utils.data
from torchvision import datasets, transforms
import torch.optim as optim
import torch.nn as nn

...全文