Bootstrap

Linux下pytorch和cuda的版本对应及安装

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
;