Tag:Just for lihui
基础内容,有些帮助
官网
https://pytorch.org/
安装方法 【开始pytorch】https://pytorch.org/get-started/locally/
文章图片
OS选择对,之后基本都是linux
CUDA版本用nvcc -V查看好,别选错
教程 https://pytorch.org/docs/stable/index.html
文章图片
有函数不知道不会用不用百度
这能搜索api 方法
还能搜文档
torch 版本与torchvision 这俩的版本要匹配
安装torchvision会自动安装torch
安装旧版本torch torchvision https://pytorch.org/get-started/previous-versions/
文章图片
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