本文已参与「新人创作礼」活动,一起开启掘金创作之路。
首先输入以下命令,先观察一下自己服务器里面的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…