pytorch的下载解决方案(下载出错、下载过慢问题)

前言

第一次下载pytorch往往会出现一些问题,比如不知道如何下载,或者下载过慢等问题,由此本文给出以下解决放方案,并给出图示解决。

正文

一、下载anaconda

首先下载anaconda,点击进入官网,如下图点击下载即可

pytorch的下载解决方案(下载出错、下载过慢问题)

二、配置虚拟环境

按照指引安装完毕后,进入配置虚拟环境

pytorch的下载解决方案(下载出错、下载过慢问题)

三、查看电脑配置

查看自己电脑的显卡型号,点击帮助中的系统信息,再转到组件,红圈标注出来的即为显卡支持的CUDA版本号,记住你的版本号,下一个步骤需要使用。

四、安装pytorch

点击进入安装选择界面

pytorch的下载解决方案(下载出错、下载过慢问题)

点击install,进入安装并选择对应的CUDA

pytorch的下载解决方案(下载出错、下载过慢问题)

pytorch的下载解决方案(下载出错、下载过慢问题)

记住此时的代码,后续会修改

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

点击anaconda的文件,进入Anaconda Prompt

pytorch的下载解决方案(下载出错、下载过慢问题)
conda activate pytorch3(你刚刚创建的环境名)

pytorch的下载解决方案(下载出错、下载过慢问题)

此时如果我们用官方的渠道下载pytorch,则有可能出现下载过慢或卡死的情况,这时我们可以转换下载源(这里用的是清华镜像源),依次输入以下代码

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

这时候拿到我们刚才的代码

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

这里得把 -c pytorch给删掉,因为它表示从官网进行下载,所以得删除,删除后得到以下命令,输入即可。

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c conda-forge

五、验证是否成功下载

按照以下步骤,输入import torch如果没有报错即为下载成功。

pytorch的下载解决方案(下载出错、下载过慢问题)

结尾

到这里就已经安装成功了,其实本文还有一些问题没有叙述完整,例如无独显的电脑下载方法,和使用pip命令直接下载等等情况的解决方案。

Original: https://blog.csdn.net/K9177/article/details/125543365
Author: K9177
Title: pytorch的下载解决方案(下载出错、下载过慢问题)

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/712012/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球