很多情況下pytorch安裝失敗都是因為網絡的問題,這里的網絡不好并不是因為自身的原因而是因為眾所周知的原因。這種情況下我們沒有辦法去解決這個總所周知的原因,但我們可以采用其他源進行pytorch安裝,或者直接離線pytorch那么怎么離線下載pytorch呢?接下來這篇文章帶你了解。
前言
每次從官網提供的下載方式,使用conda安裝pytorch的時候都會因為網速問題導致失敗。
使用conda清華源安裝也不失為一種解決方案,然而這種方式在指定cuda版本時非常不方便。
本文主要針對這一問題給出解決方案。
方法
以安裝pytorh-nightly, cuda80, linux系統(tǒng)為例。
從pytorch官網選擇 pip安裝,并選定對應的系統(tǒng), cuda版本, python版本。這時,會出現(xiàn)兩條對應的安裝命令。
第一條為安裝numpy和torchvison的命令,可直接執(zhí)行。
第二條為安裝pytorch對應的命令,直接執(zhí)行安裝速度很慢。這里我們可以采用預先下載安裝包的方式安裝pytorch。
方法如下:
在命令上輸入pytorch顯示的第二條命令
pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cu80/torch_nightly.html
命令行會顯示如下內容
其中
https://download.pytorch.org/whl/nightly/cu80/torch_nightly-1.1.0.dev20190429-cp37-cp37m-linux_x86_64.whl
是我們需要提前下載的安裝包地址。
通過迅雷下載安裝包。
將安裝包上傳至服務器
使用pip 離線安裝pytorch包
source activate python虛擬環(huán)境名 pip install /path/of/the/pytorch_package(pytorch離線安裝包路徑)
至此,大功告成!
補充:ubuntu中利用anaconda快速配置pytorch,及網絡源連接失敗問題解決
在ubuntu中利用anaconda可以方便快速的配置pytorch。
具體步驟如下:
1.安裝anaconda
bash Anconda3-x.x.x-Linux-x86_64.sh
2.創(chuàng)建python虛擬環(huán)境,這里是python3.6,環(huán)境名自擬。
conda create -n <your-virtualenv-name> python=3.6
3.激活進入虛擬環(huán)境
source activate <your-virtualenv-name>
4.加入清華源
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 --set show_channel_urls yes
5.安裝pytorch,可以根據(jù)pytorch官網信息自行匹配版本,這里選擇cuda10.1
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
6.檢核pytorch是否安裝成功可以在進入python環(huán)境之后,輸入下面命令
import torch
print(torch.__version__)
#如果返回版本信息,則安裝成功
PS:若在第5步安裝過程中存在如下所示連接網絡失敗錯誤,則刪去命令后面 -c python 再次執(zhí)行
CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/pytorch-1.3.1-py3.6_cuda10.1.243_cudnn7.6.3_0.tar.bz2> Elapsed: - An HTTP error occurred when trying to retrieve this URL. HTTP errors are often intermittent, and a simple retry will get you on your way.
以上就是pytorch安裝失敗的解決方案,希望能給大家一個參考,也希望大家多多支持W3Cschool。