Pytorch 配置 GPU 版本
博主 是 通过 Anaconda 来管理配置Python工具包的。至于Anaconda 的下载安装,比较简单,在其官网下载,基本就是一步接着一步就好~ 这里我不记录咯 ~如果大家有需求的话,留言就好 ~
------>>首先
,通过 终端进入conda 界面:
------>>其次
,为了在安装过程中速度快一点点,我们先设置下镜像源:
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:conda activate pytorch
,并且输入命令下载 符合我们自己电脑版本的 GPU版的Pytorch:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia
,注意,该命令不是随意输入的,需要根据自己的电脑 里面NVIDIA的版本。
点击 我 可以 查看符合自己版本的下载语句!!!
进去pytorch官网,下拉,可以看到如下所示的界面:(注意要 选择 和自己版本匹配的喔!!!这样就可以看到对应的语句了~)
为了下载快一点,可以把上述语句的 -c python 去掉。即 conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c nvidia
这样就好。这只是我的一个示例,具体的可以根据自己选择来。
------>>然后
!就是 漫长的安装界面了,期间 可能还需要你 输入 y
并且按回车确定,这个知道就行,操作还是比较简单的。
就比如这样:(漫 长 的 等 待 --)
安装完成后。博主再继续记录(更于 2022.11.6 21:59)