pytorch安装，解决pytorch下载报错（ERROR: Exception: Traceback (most recent call last): File ...）、下载缓慢问题

GG向前冲

于 2024-06-02 11:00:26 发布

阅读量1.3k

点赞数 10

文章标签： pytorch 人工智能 python 深度学习数据分析机器学习 pip

本文链接：https://blog.csdn.net/m0_70466650/article/details/139388058

版权

安装问题：

在下载pytorch时，由于安装包可能较大，一般可能是2G左右，通常需要花费较长的时间才能下载完成，但在很多时候，下载到一般会出现ERROR: Exception: Traceback (most recent call last): File …的错误，下面将对此提供一些解决方案。

首先进入pytorch官网（链接点击这里），根据自己的电脑配置，选择合适的版本，如果有N卡，可以选择GPU版本（需先下载N卡驱动等配置），无N卡或A卡则可以使用CPU版本，具体详细教程可以参考对应安装资料，本文简单展示。

在这里插入图片描述
在下载之前，还需进行设置国内镜像源，用以解决下载速度慢的问题：

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

该代码使用清华大学 TUNA 镜像站点来获取软件包，可以有效减少下载软件包时的延迟和速度慢的问题，提高安装软件包的效率。将代码复制到conda命令窗中即可。
在这里插入图片描述

随后将下方安装torch代码复制到conda命令窗中，本文使用采用11.6版本的安装包，该安装包可根据自己的配置进行选择。
在这里插入图片描述

pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116

在下载代码后面，可以加上

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

用以解决安装时异常的问题，将安装命定复制输入到anaconda prompt中
在这里插入图片描述
稍等片刻即可安装成功

关注