开始pytorch

Tag:Just for lihui
基础内容,有些帮助
官网
https://pytorch.org/

安装方法

https://pytorch.org/get-started/locally/
开始pytorch_第1张图片
OS选择对,之后基本都是linux
CUDA版本用nvcc -V查看好,别选错

教程

https://pytorch.org/docs/stable/index.html
开始pytorch_第2张图片
有函数不知道不会用不用百度
这能搜索api 方法
还能搜文档

torch 版本与torchvision

这俩的版本要匹配
安装torchvision会自动安装torch

安装旧版本torch torchvision

https://pytorch.org/get-started/previous-versions/
开始pytorch_第3张图片

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
torch==1.5.0+cu101
cu101意味着cuda版本是10.1

你可能感兴趣的:(学习大礼包)