pytorch安装

时间:2024-04-07 11:37:14

最近在接触深度学习方面的东西,所以难免用到pytorch框架,安装的时候遇到一些问题,幸好最终得到解决。
首先,需要去pytorch官方网站选择需要下载的版本
https://pytorch.org/get-started/locally/
这是我选择的,将run this Command 后的信息复制到控制台
pytorch安装
win+r,输入cmd回车打开控制台,输入pip install…回车
pytorch安装
你会发现,下载速度很慢,于是我切换了国内的很多开源镜像,结果还是不行,下载一会儿便会停止并且报错。后来经过一番搜索,寻得解决方案:
pytorch安装
在浏览器中打开该网址,并按照提示的版本找到需要下载的文件:
pytorch安装
这里有个小坑,就是cp37,cp38等,这里如何选择十分重要,选择错误安装不上,这个与你python的版本相关。在控制台输入python,察看版本号,如果是3.7.版本则下载cp37,如果是3.8.则需要下载cp38,以此类推。
pytorch安装
所以我选择的是:cpu/torch-1.4.0%2Bcpu-cp37-cp37m-win_amd64.whl 以及 cu101/torchvision-0.5.0-cp37-cp37m-win_amd64.whl。可以使用浏览器下载,如果下载速度也慢的话,则选择迅雷下载,鼠标右键链接,选择>复制链接地址,并在迅雷中添加下载任务即可。
下载好以后,在控制台输入 pip install 文件存放的路径,并回车,即可成功安装pytorch