您好,登錄后才能下訂單哦!
1、pytorch官網下載對應安裝文件
https://pytorch.org/
在getstarted處選擇對應版本
方法一:直接使用對應的安裝指令進行安裝
但是其實這樣比較容易斷或者出現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環境下pytorch的安裝與檢查驗證(解決runtimeerror問題),希望對大家有所幫助,如果大家有任何疑問請給我留言,小編會及時回復大家的。在此也非常感謝大家對億速云網站的支持!
如果你覺得本文對你有幫助,歡迎轉載,煩請注明出處,謝謝!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。