cuda版本以及pytorch版本选择问题

470 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路。 首先输入以下命令,先观察一下自己服务器里面的cuda版本 在这里插入图片描述

哦!如图,显然是cuda - 11.1

上pytorch官网。都到cuda 11.3了 在这里插入图片描述 进入以前pytorch版本的页面,找对应的11.1

在这里插入图片描述 往下找到了对应cuda11.1 的版本和命令了。

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f download.pytorch.org/whl/torch_s…