訴求
由於一些內網環境無法使用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