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)行源的查找，在開(kāi)始下載后停掉，然后找到屏幕上打印出來(lái)的鏈接，手動(dòng)下載whl 然后手動(dòng)安裝。
方法二：選擇相應(yīng)版本的whl進(jìn)行離線(xià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問(wèn)題)，希望對(duì)大家有所幫助，如果大家有任何疑問(wèn)請(qǐng)給我留言，小編會(huì)及時(shí)回復(fù)大家的。在此也非常感謝大家對(duì)腳本之家網(wǎng)站的支持！
如果你覺(jué)得本文對(duì)你有幫助，歡迎轉(zhuǎn)載，煩請(qǐng)注明出處，謝謝！
