由于一些内网环境无法使用pip命令安装python三方库,寻求一种能够离线安装pytorch的方法。
方法
由于是内网,首选使用Anaconda代替Python,这样无需手动配置numpy等额外依赖。
访问pytorch离线下载网址根据系统和CUDA版本选择自己需要的whl文件
一共有两个,pytorch
和torchvision
,例如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
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:Pytorch离线安装方法 - Python技术站