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