天天看点

服务器离线安装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

继续阅读