【Pytorch】pytorch和cuda版本对应及其安装命令

一个小故事

《我有个朋友》，刚入门科研信心满满，热血澎湃。有了自己小小的想法，满怀激动开始要动手实现一下。点开pycharm/jupyter，写下第一行import torch，run，然后…报错了。

网上好多博文就不要乱看了，很多都是搬运的

时效性差不说，可能还解决不了问题

直接上官网看，链接放在下面了，非常全

https://pytorch.org/get-started/previous-versions/

这里贴一小段，其他版本的大家自行查阅官网

Linux and Windows

# CUDA 10.2
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.3
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cpuonly -c pytorch