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