linux或windows環(huán)境下pytorch的安裝與檢查驗證(解決runtimeerror問題)
1、pytorch官網(wǎng)下載對應安裝文件
在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)站的支持!
如果你覺得本文對你有幫助,歡迎轉載,煩請注明出處,謝謝!
相關文章
centos中NAT模式下靜態(tài)IP連接外網(wǎng)
這篇文章主要介紹了centos中NAT模式下靜態(tài)IP連接外網(wǎng),具有一定的參考價值,感興趣的小伙伴們可以參考一下2017-06-06Ubuntu 18.04中截圖工具shutter的編輯按鈕不可用的解決辦法
Shutter是一個由第三方提供的在Ubuntu上運行的截圖工具。這篇文章主要介紹了Ubuntu 18.04中截圖工具shutter的編輯按鈕不可用的解決辦法及Ubuntu18.04 截圖工具推薦,需要的朋友可以參考下2018-08-08