开始pytorch

Tag:Just for lihui
基础内容,有些帮助
官网
https://pytorch.org/
安装方法 【开始pytorch】https://pytorch.org/get-started/locally/
开始pytorch
文章图片

OS选择对,之后基本都是linux
CUDA版本用nvcc -V查看好,别选错
教程 https://pytorch.org/docs/stable/index.html
开始pytorch
文章图片

有函数不知道不会用不用百度
这能搜索api 方法
还能搜文档
torch 版本与torchvision 这俩的版本要匹配
安装torchvision会自动安装torch
安装旧版本torch torchvision https://pytorch.org/get-started/previous-versions/
开始pytorch
文章图片

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

    推荐阅读