映象安裝pytorch的簡便方法總結

安全劍客發表於2020-04-04
PyTorch是一個開源的Python機器學習庫,基於Torch,用於自然語言處理等應用程式。2017年1月,由Facebook人工智慧研究院(FAIR)基於Torch推出了PyTorch。本文主要介紹映象安裝pytorch的簡便方法

映象安裝pytorch的簡便方法總結映象安裝pytorch的簡便方法總結

1.安裝anacoda

這網上已有許多教程,這裡不再贅述。

2.顯示卡加速(安裝CPU版本的pytorch跳過)

請原諒我在這裡又省略了,但我找了一個很好的部落格。按照這個來萬無一失,注意安裝時一定要與自己電腦版本匹配。

安裝pytorch

進入PyTorch的官網進行版本選擇

映象安裝pytorch的簡便方法總結映象安裝pytorch的簡便方法總結
注意這裡package選擇pip
例如:我選擇後的commend為

pip install torch===1.4.0 torchvision===0.5.0 -f 

這就是我們的安裝 ,而在這個安裝 之後跟隨映象設定

-i 

上面是清華的映象源,現階段已不可使用,提供下面幾個映象源用來參考。
豆瓣(douban)

清華大學

阿里雲

中國科技大學

中國科學技術大學

清華和科大的映象已經掛掉了,所以題主在這裡使用的是阿里雲的映象,最終的pip命令如下:
pip install torch===1.4.0 torchvision===0.5.0 -f -i some-package
以上,請根據自己選擇的pytorch型號和映象地址改變以上pip命令來滿足您的需要。

原文地址:

來自 “ ITPUB部落格 ” ,連結:http://blog.itpub.net/31559985/viewspace-2684509/,如需轉載,請註明出處,否則將追究法律責任。

相關文章