前言
每次从官网提供的下载方式,使用conda安装pytorch的时候都会因为网速问题导致失败。使用conda清华源安装也不失为一种解决方案,然而这种方式在指定cuda版本时非常不方便。本文主要针对这一问题给出解决方案。
方法
以安装pytorh-nightly, cuda80, linux系统为例。
从pytorch官网选择 pip
安装,并选定对应的系统
, cuda版本
, python
版本。这时,会出现两条对应的安装命令。
第一条为安装numpy
和torchvison
的命令,可直接执行。
第二条为安装pytorch对应的命令,直接执行安装速度很慢。这里我们可以采用预先下载安装包的方式安装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虚拟环境名 pip install /path/of/the/pytorch_package(pytorch离线安装包路径)
至此,大功告成!