由於訓練機無法聯網,考慮離線安裝pytorch
一、確保已安裝cuda和cudnn
這裡我安裝的是cuda9.2 + cudnn7.6.5
二、安裝pytorch1.4
pytorch的安裝一般來說是使用conda或者pip來安裝,我選擇使用pip來離線安裝,將pytorch安裝在python2.7中。
(1)檢視自己的環境適合pip安裝什麼版本的whl檔案
python
import pip._internal
print pip._internal.pep425tags.get_supported()
然後檢視適合什麼版本的whl,例如我這裡會列印出:('cp27','cp27mu','linux_x86_64'),等等。。。
(也可以從pytorch的官網進入)
(3)安裝pytorch
sudo pip install torch-1.4.0+cu92-cp27-cp27mu-linux_x86_64.whl
sudo pip install torchvision-0.5.0+cu92-cp27-cp27mu-linux_x86_64.whl
(4)安裝完成
安裝完成後開啟python測試時候安裝成功
import torch
print torch.__version__
成功安裝以後就會列印出1.4.0+cu92
(5)補充:
有可能自己的python環境中缺少一些pytorch的依賴包,導致import error。這時只要把缺少的依賴包一樣離線安裝就行了。
例如我的環境出現:importerror:no module named builtins
解壓後進入future包路徑安裝:sudo python setup.py install
安裝完成後問題解決。
PyTorch儲存和載入模型CUDA
在儲存了使用cuda訓練的模型後,載入時也一定得保持一致,換句話說,在定義網路的時候需要用 net.to device 而且在測試的時候也需要把輸入和標籤統統轉移到cuda上面,即 inputs,labels inputs.to device labels.to device 如果你希望載入使用cu...
pytorch 拓展cuda語言 多維索引問題
四維矩陣索引公式 四維矩陣 n,c,h,w 當前四維索引為 n,c,h,w,out idx n c h w c h w h w w pytorch中的 permute a torch.randn 5,3,4 a的size為 5,3,4 b a.permute 0,2,1 此時b的size會變成 5,...
PyTorch 深度學習實踐 第9講
第9講 多分類問題 源 b站 劉二大人 傳送門pytorch深度學習實踐 多分類問題 說明 1 softmax的輸入不需要再做非線性變換,也就是說softmax之前不再需要啟用函式 relu softmax兩個作用,如果在進行softmax前的input有負數,通過指數變換,得到正數。所有類的概率求...