Windows下安装PyTorch

最新推荐文章于 2024-07-27 17:33:32 发布

DiahannWu

最新推荐文章于 2024-07-27 17:33:32 发布

阅读量510

点赞数 1

文章标签： python pip 神经网络深度学习人工智能

本文链接：https://blog.csdn.net/EstelleWu/article/details/106698861

版权

本文详细介绍了在Windows10上安装PyTorch的步骤，包括如何解决因下载速度慢导致的问题，以及如何利用清华源顺利安装。通过下载安装包到本地，然后使用pip配合清华大学的镜像源进行安装，确保了安装过程的顺利完成。

摘要由CSDN通过智能技术生成

PyTorch进行神经网络的学习十分有用，但是，其在中国大陆的安装包下载十分缓慢。这里介绍一下我在Windows10中安装PyTorch的过程与建议。

1、版本选择，进入PyTorch的官网：https://pytorch.org/get-started/locally/

这里需要提醒一下，PyTorch的版本在国内下载是非常非常非常慢的，本人刚开始也是尝试了无数次的官网直接下载，最终都因为下载速度无比的慢而失败。以上图为例，如果小伙伴们想尝试直接官网下载，以管理员运行cmd，输入如下命令：

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

2、下载安装包到本地

建议大家在Package一项中选择“pip”安装，然后在“Run this Command”这一栏，直接复制代码中的两个网址，

以我上面的版本选择为例，即在新的浏览器页面中直接输入：

https://download.pytorch.org/whl/torch_stable.html

选择cpu/torch==1.5.0 cpu/torchvision==0.6.0的版本下载到本地，浏览器下载速度较慢，迅雷会稍微快点。

3、安装（这里我直接用本地下载的源失败了，