conda install 安装不了 pytorch ,后来使用 pip 解決这个问题。但是,pip时,需要增加下载的URL,才能下载。


  • pip install <包名> 或 pip install -r requirements.txt 在线安装
  • pip wheel <包名> 打包
  • pip uninstall <包名> 或 pip uninstall -r requirements.txt 卸载包

在线下载

对于cuda 9.2 版本的来说,如下

pip install torch==1.5.0+cu92 torchvision==0.6.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html

其它的cuda 版本,可以参考如下网站:
https://pytorch.org/get-started/previous-versions/


离线下载:

当然,也可以离线下载 whl,然后再安装。
https://download.pytorch.org/whl/torch_stable.html

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐