当前位置:首页 » 《随便一记》 » 正文

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

6 人参与  2023年04月03日 17:15  分类 : 《随便一记》  评论

点击全文阅读


前言

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

正文

一、下载anaconda

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

二、配置虚拟环境 

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

三、查看电脑配置 

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

四、安装pytorch

点击进入安装选择界面

 

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

 

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

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

点击anaconda的文件,进入Anaconda Prompt 

conda activate pytorch3(你刚刚创建的环境名)

输入代码进入你刚刚创建的环境  

此时如果我们用官方的渠道下载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 yesconda 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如果没有报错即为下载成功。

 结尾

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

 

 

 

 


点击全文阅读


本文链接:http://zhangshiyu.com/post/58234.html

<< 上一篇 下一篇 >>

  • 评论(0)
  • 赞助本站

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。

关于我们 | 我要投稿 | 免责申明

Copyright © 2020-2022 ZhangShiYu.com Rights Reserved.豫ICP备2022013469号-1