Pytorch 配置 GPU 版本

博主 是 通过 Anaconda 来管理配置Python工具包的。至于Anaconda 的下载安装,比较简单,在其官网下载,基本就是一步接着一步就好~ 这里我不记录咯 ~如果大家有需求的话,留言就好 ~

------>>首先,通过 终端进入conda 界面:
【简易 教程:Pytorch 配置 GPU版本】-LMLPHP
------>>其次,为了在安装过程中速度快一点点,我们先设置下镜像源:

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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro/
conda config --set show_channel_urls yes

需要指出的是,上述每一行命令,都是需要每次执行一句哈,即 你复制时,复制一行,就执行一行,(而不是把这些全部复制了,再全部粘贴!这样不行的哦!!)

另外,如果有朋友之前已经引入过镜像源了,那么会报一些 Warning,这无关紧要,请继续执行命令即可。

------>> 为了便于Pytorch的安装,我们为它单独创立一个存放其环境的地方:conda create -n pytorch python=3.9,后面的3.9指的是Python的版本,这个可以根据需要自己选择。

在输入上述命令过后,我们应该会看到如下界面:
【简易 教程:Pytorch 配置 GPU版本】-LMLPHP

------>>紧接着,我们切换环境,进入刚刚建立的Pytorch:conda activate pytorch,并且输入命令下载 符合我们自己电脑版本的 GPU版的Pytorch:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia,注意,该命令不是随意输入的,需要根据自己的电脑 里面NVIDIA的版本。

点击 可以 查看符合自己版本的下载语句!!!

进去pytorch官网,下拉,可以看到如下所示的界面:(注意要 选择 和自己版本匹配的喔!!!这样就可以看到对应的语句了~)

【简易 教程:Pytorch 配置 GPU版本】-LMLPHP

为了下载快一点,可以把上述语句的 -c python 去掉。即 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia 这样就好。这只是我的一个示例,具体的可以根据自己选择来。
【简易 教程:Pytorch 配置 GPU版本】-LMLPHP

------>>然后!就是 漫长的安装界面了,期间 可能还需要你 输入 y并且按回车确定,这个知道就行,操作还是比较简单的。

就比如这样:(漫 长 的 等 待 --)
【简易 教程:Pytorch 配置 GPU版本】-LMLPHP
安装完成后。博主再继续记录(更于 2022.11.6 21:59)

11-07 09:46