pip install 安装 torch cuda 11.8 cu118
百度了一下，有推荐这么改的，速度还是慢，因为还是转到官网 pytorch.org。最终采用 其中的版本 2.3.1，可以看看 pip 源。默认安装，速度比较慢。
·
默认安装，速度比较慢
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
百度了一下，有推荐这么改的，速度还是慢，因为还是转到官网 pytorch.org
pip install torch torchvision torchaudio --index-url https://mirror.sjtu.edu.cn/pytorch-wheels/cu118
最终采用 其中的版本 2.3.1，可以看看 pip 源
pip install torch==2.3.1+cu118 torchvision==0.18.1+cu118 torchaudio==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
欢迎来到FlagOS开发社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
10
0- 0
已为社区贡献2条内容
所有评论(0)