前言
第一次下载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命令直接下载等等情况的解决方案。