1. 根据驱动版本选择cuda版本
2. 根据cuda版本选择pytorch版本
- 进入pytorch官网:https://pytorch.org
- 选择不同的cuda版本会有不同的安装指令
- 需要安装更早的版本点击Previous versions of PyTorch根据需要查找安装指令
- 有时候用列出的指令安装总是报错,可以进入下面的链接查找对应版本,然后 wget + 文件地址 下载,pip install + 文件名 安装(直接pip install + 文件地址 好像也可以安装)
(选择下载文件安装时要注意对应自己的python版本和操作系统)
(torch和torchvision的版本对应可以参考上面同一个指令里的版本对应)
(torchvision直接pip install torchvisi