装了很多次pytorch了,但隔段时间总会遗忘或遇到些问题,此处记录一下。
安装步骤
对于安装流程很多人都写过相关blog,此处不在赘述。下面贴一下参考链接:
- Windows安装Pytorch-GPU版本
- PyTorch-GPU 安装之 conda install & pip install 区别详解
- 服务器RTX3090安装显卡驱动,cuda,cudnn,gpu版pytorch记录
- Linux安装Pytorch1.8GPU(CUDA11.1)的实现
常用网址
- torch,torchvision离线下载地址
- torch, torchvision, python 的版本对应关系
- cuda与显卡驱动的 版本对应关系
- NVIDIA显卡驱动下载地址
- cuda下载地址
- cudnn下载地址
- pytorch官网
疑问解答
- 显卡驱动, cuda, cudnn的关系与版本对应问题
- 本机已有显卡驱动和cuda,直接装pytorch即可。不必在重新装驱动和cuda,除非你想换其他版本的驱动。
- conda或pip装 pytorch指定版本时,若服务器端找不到对应可下载的库,则会默认安装CPU版本的torch。