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

https://pytorch.org/

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

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

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

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

找到下載鏈接

進(jìn)行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)載，煩請注明出處，謝謝！