Pytorch离线安装方法


诉求

由于一些内网环境无法使用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


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM