由于一些内网环境无法使用pip命令安装python三方库,寻求一种能够离线安装pytorch的方法。

方法

由于是内网,首选使用Anaconda代替Python,这样无需手动配置numpy等额外依赖。

访问pytorch离线下载网址根据系统和CUDA版本选择自己需要的whl文件

一共有两个,pytorchtorchvision,例如win10x64下cuda10.1的python3.7对应下载文件为

cu101/torch-1.5.0%2Bcu101-cp37-cp37m-win_amd64.whl
cu101/torchvision-0.6.0%2Bcu101-cp37-cp37m-win_amd64.whl

源址下载较慢可用迅雷等P2P工具替代之

之后是离线安装,打开Anaconda Navigator - Powershell Prompt输入

(base) PS E:\cu101> pip install .\torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl .\torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl

注意顺序,后装的vision依赖torch,中途会自动搜寻并发现满足的numpy,future,pillow等依赖。

验证

至此,安装完毕,使用torch自带函数验证是否成功。

>>> import torch
>>> torch.cuda.is_available()
True