pip 安装 cuda 9.2 版本的 Pytorch
conda install 安装不了 pytorch ,后来使用 pip 解決这个问题。但是,pip时,需要增加下载的URL,才能下载。pip install <包名> 或 pip install -r requirements.txt在线安装pip wheel <包名>打包pip uninstall <包名> 或 pip uninstall -r requir
·
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
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)