鱼C论坛

 找回密码
 立即注册
查看: 2073|回复: 1

[技术交流] 如何在windows下安装 torch

[复制链接]
发表于 2020-6-20 01:13:08 | 显示全部楼层 |阅读模式

马上注册,结交更多好友,享用更多功能^_^

您需要 登录 才可以下载或查看,没有账号?立即注册

x
本帖最后由 txxcat 于 2020-6-20 01:13 编辑

    torch在windows下用pip直接安装不了,linux和mac os可能可以,windows使用conda安装会方便点,如果想在python下自己安装,就需要去官网找了:https://pytorch.org/get-started/locally/
    官网提供了各种选项,按需选择后官网提供了安装命令,比如我没有NVidia的显卡,不能使用CUDA,CUDA选择了None,最后给出这个命令:
  
pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
    很遗憾,运行报错,说没有找到合适版本,可能是bug吧,那就自己找找,打开命令里的地址:
    https://download.pytorch.org/whl/torch_stable.html
    众多的版本,终于找到一个:cpu/torch-1.5.0%2Bcpu-cp38-cp38-win_amd64.whl,和:cpu/torchvision-0.6.0%2Bcpu-cp38-cp38-win_amd64.whl,可以点下面地址直接下载,不用自己找,记得要下两个文件,torchvision要小多了:
    https://download.pytorch.org/whl ... -cp38-win_amd64.whl
    https://download.pytorch.org/whl ... -cp38-win_amd64.whl
    不过速度嘛,比百度网盘非VIP还慢,104M估摸要一、两个小时吧,原因就不说了。然后安装就OK了:
  
pip install torch-1.5.0+cpu-cp38-cp38-win_amd64.whl
pip install torchvision-0.6.0+cpu-cp38-cp38-win_amd64.whl
    如果你有NVidia的显卡,弄清楚CUDA的版本,根据版本找对应的whl文件下载,运行效率增加的可不止一点点。
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复

使用道具 举报

发表于 2020-10-25 12:00:50 | 显示全部楼层
刚用到了 厉害
想知道小甲鱼最近在做啥?请访问 -> ilovefishc.com
回复 支持 反对

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

小黑屋|手机版|Archiver|鱼C工作室 ( 粤ICP备18085999号-1 | 粤公网安备 44051102000585号)

GMT+8, 2025-1-18 05:26

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

快速回复 返回顶部 返回列表