参考:

https://wenku.csdn.net/column/gxg4d619i12

根据我的cuda版本,安装

pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐