天天看點

伺服器離線安裝Pytorch 1.01以及torchvision 0.2.1

由于伺服器沒有使用代理,直接通過線上安裝pytorch的速度實在是太慢。是以我們考慮離線安裝pytorch。

1. 伺服器環境

  1. Ubuntu: 14.04 64bit
  2. Anaconda: 4.5.4
  3. CUDA: 8.0.61
  4. cuDNN: 5.1.5
  5. gcc: 4.8.4

2. 離線安裝

在pytorch官網我們可以很友善的查找安裝指令:

伺服器離線安裝Pytorch 1.01以及torchvision 0.2.1

這裡我們一開始使用

conda

指令

conda install pytorch torchvision cudatoolkit=8.0 -c pytorch

發現不行,然後使用線上pip3安裝:

pip3 install https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision
           

發現速度實在太慢,有點拖節奏。 于是我們考慮離線安裝。

pytorch和pytorchvision安裝包,我們可以把他本地下載下傳下來。

  1. torch1.01: https://download.pytorch.org/whl/cu80/torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
  2. torchvision0.2.1: https://pypi.org/project/torchvision/0.2.1/#files

然後通過

scp

指令将本地檔案上傳到伺服器指定位置。

然後在伺服器上進行安裝,在控制台輸入指令:

pip3 install torch-1.0.1.post2-cp36-cp36m-linux_x86_64.whl
pip3 install torchvision-0.2.1-py2.py3-none-any.whl
           

這樣就安裝好啦。然後我們測試一下:

伺服器離線安裝Pytorch 1.01以及torchvision 0.2.1

繼續閱讀