1、pytorch官網(wǎng)下載對應(yīng)安裝文件
https://pytorch.org/
在getstarted處選擇對應(yīng)版本
方法一：直接使用對應(yīng)的安裝指令進行安裝
但是其實這樣比較容易斷或者出現(xiàn)runtimeout的error。解決辦法：使用pip先進行源的查找，在開始下載后停掉，然后找到屏幕上打印出來的鏈接，手動下載whl 然后手動安裝。
方法二：選擇相應(yīng)版本的whl進行離線下載
找到下載鏈接
進行whl文件的下載然后安裝
pytorch從鏈接安裝指定版本
pip3 install whl
pip3 install torchvision
2、驗證
import torch
import torchvision
print(torch.cuda.is_available())#輸出為true即可
3、pytorch版本查看
import torch
print(torch.__version__)
總結(jié)
以上所述是小編給大家介紹的linux或windows環(huán)境下pytorch的安裝與檢查驗證(解決runtimeerror問題)，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復(fù)大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！
如果你覺得本文對你有幫助，歡迎轉(zhuǎn)載，煩請注明出處，謝謝！
