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

https://pytorch.org/

在getstarted處選擇對應版本

方法一：直接使用對應的安裝指令進行安裝

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

方法二：選擇相應版本的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__)

總結

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