txxcat 发表于 2020-6-20 01:13:08

如何在windows下安装 torch

本帖最后由 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/cpu/torch-1.5.0%2Bcpu-cp38-cp38-win_amd64.whl
    https://download.pytorch.org/whl/cpu/torchvision-0.6.0%2Bcpu-cp38-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文件下载,运行效率增加的可不止一点点。

zhangfeiwen 发表于 2020-10-25 12:00:50

刚用到了 厉害
页: [1]
查看完整版本: 如何在windows下安装 torch