在Windows下安裝配置CPU版的PyTorch的方法

由于我已經(jīng)安裝了anaconda，所以不在贅述，下載可以上清華鏡像版下載 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 我下的版本為Anaconda3-5.2.0-Windows-x86_64 對(duì)應(yīng)python版本為3.6.5 版本根據(jù)個(gè)人需求下載即可。下面開(kāi)始具體的PyTorch的cpu版本安裝。

1.添加鏡像源

還是使用清華源下載，打開(kāi)cmd或者 anaconda prompt,輸入以下代碼：

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/pytorch/
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/msys2/
Conda config --set show_channel_urls yes

在這里，我們配置了pkgs/free/，pkgs/main/和cloud/pytorch/等好幾個(gè)清華源下的目錄，以便為pytorch所需的全部相關(guān)庫(kù)提供國(guó)內(nèi)鏡像源。
添加完畢后，運(yùn)行下面這段代碼查看鏡像源是否設(shè)置成功：

conda config --show

如果在channels下方可以查看到我們添加的鏡像源，那就是設(shè)置成功了。如下圖：

2.在Win10下配置PyTorch（CPU版）

打開(kāi)cmd，輸入以下代碼查看當(dāng)前環(huán)境：

conda info -e

運(yùn)行結(jié)果如下所示：

可以看到當(dāng)前環(huán)境只有一個(gè)base環(huán)境，然后輸入命令安裝一個(gè)python：

conda create -n torch python=3.6

版本可以自己選擇，遇到選擇Y/N的，一律選擇Y就行。安裝完成如下圖：

安裝完成后，輸入下行命令進(jìn)入環(huán)境

conda activate torch

如圖，出現(xiàn)（torch）即進(jìn)入成功，然后進(jìn)入百度https://pytorch.org/進(jìn)入PyTorch官網(wǎng)，點(diǎn)擊Get Started，進(jìn)入后如下選擇：

由于安裝的是CPU版本，所以不需要cuda，選擇完成后，會(huì)出現(xiàn)上圖紅框中的代碼，復(fù)制粘貼到cmd命令中：

conda install pytorch torchvision torchaudio cpuonly

回車后就會(huì)進(jìn)行安裝（安裝過(guò)程需要一段時(shí)間，所以可以打把游戲啥的）老規(guī)矩，遇Y/N，無(wú)腦Y就行。完成后如下圖

到這步，基本算是安裝成功了，只需要簡(jiǎn)單的驗(yàn)證即可。
進(jìn)入python，輸入import torch，如果沒(méi)有報(bào)錯(cuò)，那就是安裝成功！

到此這篇關(guān)于在Windows下安裝配置CPU版的PyTorch的文章就介紹到這了,更多相關(guān)Windows配置CPU版的PyTorch內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！