Windows安装Pytorch,以及踩过的坑

507 阅读1分钟

首先是安装流程

通过镜像源安装Anaconda,清华的镜像源:

mirrors.tuna.tsinghua.edu.cn/anaconda/ar…

安装好Anaconda之后,上Pytorch官网上获取对应的Pytorch版本下载 这里遇到第一个小坑,CUDA版本的选择

CUDA是适配英伟达的N卡的,没有N卡这个就应该选择none,如果是N卡,通过N卡的控制面板查看CUDA对应的版本 控制面板——>帮助——>系统信息——>组件。我这里是10.1版本就选择10.1的CUDA

最大的坑

很多教程没有明确官网的指令下载默认用外网的源,即使你已经添加了清华的源**

指令里面的 -c pytorch去掉才可以通过国内的镜像下载

还是有坑

pytorch和CUDAtoolkit会下载一半突然中止,然后报错CondaHTTPError:HTTP 000 CONNECTION FAILED**

提示让我retry,但是retry几次都没用,Anaconda prompt越跑越卡,最后占满内存炸掉了,弄了半天这个问题没有解决,一怒之下重装了Anaconda,现在看来估计是没有clean cache

conda clean -y -all //删除所有的安装包及cache

重装之后还是遇到这个问题,还是会卡掉,换了几次源,结果差不多,最后又重试了几遍才成了

应该是网络波动和镜像那边服务器的网络问题,一旦断了网就不继续下了

这里给出最后用到的镜像

channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

ssl_verify: true
show_channel_urls: true
report_errors: true

最后一个坑

numpy版本和pip版本过老的问题,会造成import torch报错

python -m pip install --upgrade pip

pip install --upgrade numpy

更新一下,即可解决