Pytorch1 4 0 cuda9 2離線安裝

2021-10-05 02:19:08 字數 952 閱讀 1232

由於訓練機無法聯網,考慮離線安裝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有負數,通過指數變換,得到正數。所有類的概率求...