更多細(xì)節(jié)請(qǐng)參考:
使用conda/pip安裝pytorch(mac/windows/linux)
PyTorch官網(wǎng)教程
本文以Linux中-pytorch版本1.0.0-清華源為-cuda90為例:
Cuda官網(wǎng)教程
同時(shí)請(qǐng)注意檢查依賴庫(kù)安裝:
pytorch安裝依賴庫(kù)
在安裝過(guò)程中遇到的問(wèn)題责掏,會(huì)反饋更新在:
pytorch安裝錯(cuò)誤匯總
(Linux/windows/Mac OS都相近十兢,除去源碼安裝的部分細(xì)節(jié)以及Mac OS不支持CUDA)
1.pip安裝pytorch:
(若為python則用pip,若為python3則用pip3)
sudo pip3 install torch==1.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
安裝過(guò)程中小白可以都用一下sudo李茫,以免出現(xiàn)“Permission Deny”等問(wèn)題字逗,當(dāng)然具體情況具體分析尘执。
沒有pip或者python請(qǐng)點(diǎn)擊傳送門:
pip安裝教程
python安裝教程
補(bǔ)充
也可以直接默認(rèn)安裝最新的pytorch的GPU版本:
sudo pip3 install torch torchvision
pip清華源升級(jí)pytorch:
sudo pip3 install --upgrade torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
pip降級(jí)安裝(以降級(jí)到torch 0.4.1為例):
方法一(簡(jiǎn)單暴力)
sudo pip3 install torch==0.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple>
方法二(容易出錯(cuò))
sudo pip3 install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl
pip3 install torchvision
(注意其中的cu90/torch-0.4.1-cp35-cp35m-win_amd64.whl草姻,會(huì)隨你的cuda, python, 版本需求改變)
pip卸載pytorch :
sudo pip3 uninstall torch
事實(shí)個(gè)人發(fā)現(xiàn),'-i' 后的鏡像網(wǎng)址輸錯(cuò)一點(diǎn)氛改,有時(shí)候也會(huì)更正帐萎。
2.conda 安裝pytorch
cuda 90下安裝pytorch:
conda install pytorch torchvision cuda90 -c soumith
pytorch 無(wú)cuda版本:
conda install pytorch-cpu torchvision-cpu -c pytorch
conda卸載pytorch
conda uninstall pytroch