整体思路为 先根据 nvidia-smi 指令 看看自己的cuda版本,然后安装对应的cuda驱动 对应关系表如下:
接着,根据安装的cuda以及python版本,安装对应的torch,cuda、python与torch对应版本号如下:
然后就可以安装torchvision了:
整体思路为 先根据 nvidia-smi 指令 看看自己的cuda版本,然后安装对应的cuda驱动 对应关系表如下:
接着,根据安装的cuda以及python版本,安装对应的torch,cuda、python与torch对应版本号如下:
然后就可以安装torchvision了: