1、pytorch官網(wǎng)下載對(duì)應(yīng)安裝文件

https://pytorch.org/

在getstarted處選擇對(duì)應(yīng)版本

方法一：直接使用對(duì)應(yīng)的安裝指令進(jìn)行安裝

但是其實(shí)這樣比較容易斷或者出現(xiàn)runtimeout的error。解決辦法：使用pip先進(jìn)行源的查找，在開始下載后停掉，然后找到屏幕上打印出來(lái)的鏈接，手動(dòng)下載whl 然后手動(dòng)安裝。

方法二：選擇相應(yīng)版本的whl進(jìn)行離線下載

找到下載鏈接

進(jìn)行whl文件的下載然后安裝

pytorch從鏈接安裝指定版本

pip3 install whl
pip3 install torchvision

2、驗(yàn)證

import torch
import torchvision
print(torch.cuda.is_available())#輸出為true即可

3、pytorch版本查看

import torch
print(torch.__version__)

總結(jié)

以上所述是小編給大家介紹的linux或windows環(huán)境下pytorch的安裝與檢查驗(yàn)證(解決runtimeerror問題)，希望對(duì)大家有所幫助，如果大家有任何疑問請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！
如果你覺得本文對(duì)你有幫助，歡迎轉(zhuǎn)載，煩請(qǐng)注明出處，謝謝！